《字典树》数组模版

来源:互联网 发布:淘宝直通车上分 编辑:程序博客网 时间:2024/05/21 09:33
#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;}

0 0
原创粉丝点击