题目描述
贝茜正在领导奶牛们逃跑．为了联络，奶牛们互相发送秘密信息．
信息是二进制的，共有M(1≤M≤50000)条．反间谍能力很强的约翰已经部分拦截了这些信息，知道了第i条二进制信息的前bi(l《bi≤10000)位．他同时知道，奶牛使用N(1≤N≤50000)条密码．但是，他仅仅了解第J条密码的前cj(1≤cj≤10000)位．
对于每条密码J，他想知道有多少截得的信息能够和它匹配．也就是说，有多少信息和这条密码有着相同的前缀．当然，这个前缀长度必须等于密码和那条信息长度的较小者．
在输入文件中，位的总数（即∑Bi+∑Ci）不会超过500000.
输入
第1行输入N和M，之后N行描述秘密信息，之后M行描述密码．每行先输入一个整数表示信息或密码的长度，之后输入这个信息或密码．所有数字之间都用空格隔开．
输出
共M行，输出每条密码的匹配信息数．
样例输入
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
样例输出
1 3 1 1 2
题解
先把信息建成字典树。每个信息的结尾打上标记，并记录以该节点结尾的有多少个。查询的时候，如果是前缀等于信息的情况，那么就返回在字典树上经过的标记的数目，如果是前缀等于密码的情况，那么返回密码的最后一个字符的子树中的标记数。
#include<cmath> #include<cstdio> #include<cstdlib> #include<cstring> #include<iostream> #include<algorithm> using namespace std; #define ll long long const int maxn=50000*10; int t[maxn][2],n,m,len,root,tot,cnt[maxn],num[maxn]; char c[500000+50]; bool word[maxn]; void insert(char *s,int r){ root=0;int id; for(int i=0;i<r;i++){ id=s[i]-'0'; if(!t[root][id]) t[root][id]=++tot; cnt[root]++;root=t[root][id]; } word[root]=true;num[root]++; } int find(char *s,int r){ root=0;int id,ans=0; for(int i=0;i<r;i++){ id=s[i]-'0'; if(!t[root][id]){ return ans; } root=t[root][id];if(word[root]) ans+=num[root]; } return ans+cnt[root]; } template<typename T>void read(T& aa){ char cc; ll ff;aa=0;cc=getchar();ff=1; while((cc<'0'||cc>'9')&&cc!='-') cc=getchar(); if(cc=='-') ff=-1,cc=getchar(); while(cc>='0'&&cc<='9') aa=aa*10+cc-'0',cc=getchar(); aa*=ff; } int main(){ read(m),read(n); for(int i=1;i<=m;i++){ int x,w; read(w);for(int i=1;i<=w;i++) read(x),c[i-1]=x+'0'; insert(c,w); } for(int i=1;i<=n;i++){ int x,w; read(w);for(int i=1;i<=w;i++) read(x),c[i-1]=x+'0'; printf("%d ",find(c,w)); } return 0; }