题目
原题来自: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;
}
当密码出现与信息不符合的时候,直接返回它的信息前缀数,因为出现不符合的情况时,该密码就不能作为前缀了。