Secret Message 秘密信息【Trie】

题目

原题来自:USACO 2008 Dec. Gold

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

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

对于每条密码 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;
}

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

上一篇:ASP.NET MVC 过滤器(三)


下一篇:.NET MVC中的过滤器