【AC自动机】HDU中模板题

【HDU2222】

最纯粹的裸题，错误点详见注释。

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<queue>
 using namespace std;
 struct ACauto
 {
     int sum;
     ACauto* next[];
     ACauto* fail;
     ACauto()
     {
         sum=;
         for (int i=;i<;i++) next[i]=NULL;
         fail=NULL;
     }
 };

 void insert(ACauto* root,char* str)
 {
     int len=strlen(str);
     ACauto* now=root;
     for (int i=;i<len;i++)
     {
         int index=str[i]-'a';
         if (now->next[index]==NULL)
         {
             ACauto* tmp=new ACauto;
             now->next[index]=tmp;
         }
         now=now->next[index];
     }
     now->sum++;
 }

 void build(ACauto* root)
 {
     queue<ACauto*> que;
     que.push(root);
     while (!que.empty())
     {
         ACauto* tmp=que.front();que.pop();
         for (int i=;i<;i++)
         {
             if (tmp->next[i]==NULL) continue;
             if (tmp==root)
                 tmp->next[i]->fail=root;
             else
             {
                 ACauto* p=tmp->fail;
                 while (p!=NULL)
                 {
                     if (p->next[i]!=NULL)
                     {
                         tmp->next[i]->fail=p->next[i];
                         break;
                     }
                     p=p->fail;
                 }
                 if (p==NULL) tmp->next[i]->fail=root;
             }
             que.push(tmp->next[i]);
         }
     }
 }

 int query(ACauto* root,char* str)
 {
     int ans=,len=strlen(str);
     ACauto* p=root;
     for (int i=;i<len;i++)
     {
         int index=str[i]-'a';
         while (p->next[index]==NULL && p!=root) p=p->fail;
         /*错误点：上述语句是while语句不是if语句*/
         p=p->next[index];
         p=(p==NULL)?root:p;
         ACauto* tmp=p;
         while (tmp!=root)
         /*-1表示这个单词之前已经被统计过了，不再重复计算*/
         {
             if (tmp->sum>=)
             {
                 ans+=tmp->sum;
                 tmp->sum=-;
             }
             else
             break;
             tmp=tmp->fail;
         }
     }
     return ans;
 }

 void submain()
 {
     ACauto* root=new ACauto;
     int n;
     char str[+];
     scanf("%d",&n);
     for (int i=;i<n;i++)
     {
         scanf("%s",str);
         insert(root,str);
     }
     build(root);
     scanf("%s",str);
     cout<<query(root,str)<<endl;
 }

 int main()
 {
     int T;
     scanf("%d",&T);
     for (int kase=;kase<T;kase++) submain();
     return ;
 }