bzoj1590 Secret Message

Description

贝茜正在领导奶牛们逃跑．为了联络，奶牛们互相发送秘密信息．

信息是二进制的，共有M(1≤M≤50000)条．反间谍能力很强的约翰已经部分拦截了这些信息，知道了第i条二进制信息的前bi(l《bi≤10000)位．他同时知道，奶牛使用N(1≤N≤50000)条密码．但是，他仅仅了解第J条密码的前cj(1≤cj≤10000)位．

对于每条密码J，他想知道有多少截得的信息能够和它匹配．也就是说，有多少信息和这条密码有着相同的前缀．当然，这个前缀长度必须等于密码和那条信息长度的较小者．

在输入文件中，位的总数（即∑Bi+∑Ci）不会超过500000.

Input

第1行输入N和M，之后N行描述秘密信息，之后M行描述密码．每行先输入一个整数表示信息或密码的长度，之后输入这个信息或密码．所有数字之间都用空格隔开．

Output

共M行，输出每条密码的匹配信息数．

Sample Input

4 5

3 0 1 0

1 1

3 1 0 0

3 1 1 0

1 0

1 1

2 0 1

5 0 1 0 0 1

2 1 1

Sample Output

1

3

1

1

2

​ 01 \(Trie\)的题，我不会。

​ 经过\(Aswert\)的讲解，我会了。

​ \(siz\)数组表示这个点的子树有多少个信息，假设一个密码比较短，它找不到\(Tire\)树的结尾，那么这颗子树下面（包括当前点）的所有信息都应该计入答案，然后在减去当前点是否是一个信息。

#include <iostream>
#include <cstdio>
#include <cctype>

using namespace std;

inline long long read() {
    long long s = 0, f = 1; char ch;
    while(!isdigit(ch = getchar())) (ch == '-') && (f = -f);
    for(s = ch ^ 48;isdigit(ch = getchar()); s = (s << 1) + (s << 3) + (ch ^ 48));
    return s * f;
}

const int N = 5e5 + 5;
int n, m, k, tot;
int a[N], t[N][2], siz[N], ans[N], flag[N];

void ins() {
    int p = 0;
    for(int i = 1;i <= k; i++) {
        int num = a[i];
        if(t[p][num] == 0) t[p][num] = ++tot;
        siz[t[p][num]]++;
        p = t[p][num];
    }
    flag[p]++;
}

int find() {
    int p = 0, res = 0;
    for(int i = 1;i <= k; i++) {
        int num = a[i];
        // cout << res << " " << t[p][num] << endl;
        if(t[p][num]) {
            p = t[p][num];
            res += flag[p];
        }
        else return res;
    }
    return res + siz[p] - flag[p];
}

void init() {
    n = read(); m = read();
    for(int i = 1;i <= n; i++) {
        k = read();
        for(int j = 1;j <= k; j++) a[j] = read();
        ins();
    }
    for(int i = 1;i <= m; i++) {
        k = read();
        for(int j = 1;j <= k; j++) a[j] = read();
        ans[i] = find();
    }
}

void pinf() {
    for(int i = 1;i <= m; i++) printf("%d\n", ans[i]);
}

int main() {

    init();
    pinf();

    return 0;
}

​