《字典树》数组模版

最新推荐文章于 2022-01-23 15:52:06 发布

ACM_sjtsjt

最新推荐文章于 2022-01-23 15:52:06 发布

阅读量1k

点赞数

本文链接：https://blog.csdn.net/a709743744/article/details/50589818

版权

#include <cstdio>
#include <cstring>
using namespace std;
const int N=1000005;
int trie[N][26],cnt[N],flag[N],num;
void insert(char *s,int id)
{
    int u=0,i=0,k;
    while(s[i])
    {
        k=s[i++]-'a';
        if(trie[u][k]==0)
            trie[u][k]=num++;
        u=trie[u][k];
        if(flag[u]!=id)
            cnt[u]++;
        flag[u]=id;
    }
}
int search(char *s)
{
    int u=0,i=0,k;
    while(s[i])
    {
        k=s[i++]-'a';
        if(!trie[u][k])
            return 0;
        u=trie[u][k];
    }
    return cnt[u];
}
int main()
{
    num=1;
    int n;
    char s[30];
    scanf("%d", &n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s", s);
        for(int j=0;s[j];j++)
            insert(s+j,i);
    }
    scanf("%d", &n);
    while(n--)
    {
        scanf("%s",s);
        printf("%d\n",search(s));
    }
    return 0;
}