Secret Message 秘密信息【Trie】

题目

原题来自：USACO 2008 Dec. Gold

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

信息是二进制的，共有 $M$条。反间谍能力很强的约翰已经部分拦截了这些信息，知道了第 i 条二进制信息的前 $b$i 位。他同时知道，奶牛使用 $N$ 条密码。但是，他仅仅了解第 j 条密码的前 $c$j位。

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

输入格式

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

输出格式

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

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

Output

1
3
1
1
2

4条信息，5条密码截获的信息前缀是 010,1,100,110，可能的密码前缀是 0,1,01,01001,11。

0只配对 010；
1配对 1,100,110；
01只配对 010；
01001配对 010；
11配对 1,110。

代码

#include <iostream>

using namespace std;

const int N=500010;

int son[N][2], cnt[N], idx, en[N];

void insert(int *str, int k){
	int p=0;
	for(int i=0; i<k; i++){
		int u=str[i];
		cnt[p]++;
		if(!son[p][u]) son[p][u]=++idx;
		p=son[p][u];
	}
	en[p]+=1;
}

int query(int *str, int k){
	int p=0, c=0;
	for(int i=0; i<k; i++){
		int u=str[i];
		if(!son[p][u]) return c;
		p=son[p][u];
		if(en[p]>0)c+=en[p];
	}
	return cnt[p]+c;
}

int s[10005];
int main(){
	int n, m, k;
	cin>>n>>m;
	while(n--){
		scanf("%d", &k);
		for(int i=0; i<k; i++){
			scanf("%d", &s[i]);
		}
		insert(s, k);
	}
	while(m--){
		scanf("%d", &k);
		for(int i=0; i<k; i++){
			scanf("%d", &s[i]);
		}
		printf("%d\n",query(s, k));
	}
	return 0;
}

当密码出现与信息不符合的时候，直接返回它的信息前缀数，因为出现不符合的情况时，该密码就不能作为前缀了。