洛谷 2922 BZOJ 1590 [USACO08DEC]秘密消息Secret Message

【题意概述】

　　给出n个01串组成的字典和m个询问，每次询问某个01串和多少个字典中的串有相同的前缀。（前缀长度是两串中较小的部分）

【题解】

　　直接上Trie树即可。树上每个节点记录两个信息：这个节点有多少个串经过，这个节点是多少个串的结尾。

#include<cstdio>
#include<algorithm>
#include<cstring>
#define LL long long
#define rg register
#define N 500010
using namespace std;
int n,m,a[N][4],now,ans,tot;
inline int read(){
    int k=0,f=1; char c=getchar();
    while(c<'0'||c>'9')c=='-'&&(f=-1),c=getchar();
    while('0'<=c&&c<='9')k=k*10+c-'0',c=getchar();
    return k*f;
}
int main(){
    memset(a,0,sizeof(a)); 
    n=read(); m=read();
    for(rg int i=1;i<=n;i++){
        int x=read(); now=0;
        while(x--){
            int y=read();
            if(a[now][y]>0) now=a[now][y];
            else a[now][y]=++tot,now=a[now][y];
            a[now][2]++;
        }
        a[now][3]++; 
    }
    while(m--){
        int x=read(); bool ok=1; ans=now=0;
        while(x--){
            int y=read();
            if(a[now][y]&&ok){
                now=a[now][y];
                if(a[now][3]) ans+=a[now][3];
            }
            else ok=0;
        }
        if(ok) ans+=a[now][2]-a[now][3];
        printf("%d\n",ans);
    } 
    return 0;
}