首页 > 代码库 > hdu2222

hdu2222

#include<iostream>#include<cstdio>#include<algorithm>#include<cstring>#include<queue>using namespace std;#define N 26#define maxn 500005struct Trie{    int next[500010][26],fail[500010],end[500010];    int root,L;    int newnode()    {        for(int i=0;i<26;i++)        next[L][i]=-1;        end[L++]=0;        return L-1;    }    void init()    {        L=0;        root=newnode();    }    void insert(char buf[])    {        int len=strlen(buf);        int now=root;        for(int i=0;i<len;i++)        {            if(next[now][buf[i]-a]==-1)            next[now][buf[i]-a]=newnode();            now=next[now][buf[i]-a];        }        end[now]++;    }    void build()    {        queue<int> Q;        fail[root]=root;        for(int i=0;i<26;i++)        {            if(next[root][i]==-1)            {                next[root][i]=root;            }else            {                fail[next[root][i]]=root;                Q.push(next[root][i]);            }        }        while(!Q.empty())        {            int now=Q.front();            Q.pop();            for(int i=0;i<26;i++)            {                if(next[now][i]==-1)                {                    next[now][i]=next[fail[now]][i];                } else                {                    fail[next[now][i]]=next[fail[now]][i];                    Q.push(next[now][i]);                }            }             }    }    int query(char buf[])    {        int len=strlen(buf);        int now=root;        int res=0;        for(int i=0;i<len;i++)        {            now=next[now][buf[i]-a];            int temp=now;            while(temp!=root)            {                res+=end[temp];                end[temp]=0;                temp=fail[temp];            }        }        return res;    }};Trie ac;int T,sz,ans,n;char buf[2*maxn];int main(){    scanf("%d",&T);    while(T--)    {        sz=1,ans=0;        scanf("%d",&n);        ac.init();         while(n--)        {            scanf("%s",buf);            ac.insert(buf);             }              ac.build();         scanf("%s",buf);         printf("%d\n",ac.query(buf));    }    return 0; } 

 

hdu2222