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)的讲解,我会了。
把每一条信息放到一颗Trie树里,(siz)数组表示这个点的子树有多少条信息。假设一个密码比较短,它找不到(Tire)树的叶子结点,找到了树上的节点(x),那么(x)的子树(包括(x))的所有信息都应该计入答案(加上(siz[x])),然后在减去当前点是否是一个信息(减去(flag[x]))。
#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
", ans[i]);
}
int main() {
init();
pinf();
return 0;
}