洛谷 [USACO08DEC]Secret Message G（01字典树）

传送门

---

解题思路

把每一个信息插入01字典树中，用vis[i]表示有多少个信息经过这个点，用vis2[i]表示有多少信息以i这个点结尾。

注意vis中不包含以其结尾的点。

然后对于每一条暗号，跑字典树，经过的点加上vis2[now]，结束时加上vis[now]，注意当因为匹配不到结束时不能加vis，只有到长度len时加上vis。

感觉说的不是很清楚，看代码吧。

AC代码

#include<cstdio>
#include<iostream>
#include<cstring>
#include<iomanip>
#include<cmath>
using namespace std;
const int maxn=500005;
int n,m,a[maxn],ch[maxn][3],vis[maxn],cnt=1,len,vis2[maxn];
void insert(int len){
    int now=1;
    for(int i=1;i<=len;i++){
        if(!ch[now][a[i]]) ch[now][a[i]]=++cnt;
        now=ch[now][a[i]];
        vis[now]++;
    }
    vis2[now]++;
    vis[now]--;//注意 
    return;
}
void query(int len){
    int now=1,ans=0;
    for(int i=1;i<=len;i++){
        if(ch[now][a[i]]) now=ch[now][a[i]];
        else{
            ans-=vis[now];//注意 
            break;
        }
        ans+=vis2[now];
    }
    ans+=vis[now];
    printf("%d\n",ans);
    return;
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        cin>>len;
        for(int j=1;j<=len;j++){
            scanf("%d",&a[j]);
        }
        insert(len);
    }
    for(int i=1;i<=m;i++){
        cin>>len;
        for(int j=1;j<=len;j++){
            scanf("%d",&a[j]);
        }
        query(len);
    }
    return 0;
}