hdu 3065 病毒侵袭持续中 trie

#include<bits/stdc++.h>
using namespace std;

struct trie
{
    int num;
    trie *next[26];
};
char s[1024][55];
int cnt[1024];
trie *rt=new trie;
char str[2000000+5];
void settrie(char *s,int num)
{
    trie *p=rt,*t;
    int n=strlen(s),i,c;
    for(i=0;i<n;i++)
    {
        c=s[i]-'A';
       if(p->next[c]==NULL)
       {
           t=new trie;
           for(int j=0;j<26;j++)
            {
                t->next[j]=NULL;
                t->num=-1;
            }
           p->next[c]=t;
       }
       if(i==n-1) p->num=num;
       p=p->next[c];
    }
}

int main()
{
    int n,i,j,len,c;
    trie *p;
    while(~scanf("%d",&n))
    {
        memset(cnt,0,sizeof(cnt));
        rt->num=-1;
        for(i=0; i<26; i++)
        {
            rt->next[i]=NULL;
        }
        for(i=0; i<n; i++)
        {
            scanf("%s",s[i]);
            settrie(s[i],i);
        }
        scanf("%s",str);
        len=strlen(str);
        for(i=0;i<len;i++)
            if('A'<=str[i]&&str[i]<='Z')
            {
                p=rt;
                j=i;
                c=str[j]-'A';
                while(p->next[c])
                {
                    if(p->num!=-1) cnt[p->num]++;
                    p=p->next[c];
                    j++;
                    if('A'<=str[j]&&str[j]<='Z') c=str[j]-'A';
                    else break;
                }
            }
        for(i=0; i<n; i++)
            if(cnt[i])
                printf("%s: %d\n",s[i],cnt[i]);
    }
    return 0;
}