【AC自动机】HDU中模板题

【HDU2222】

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

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

void insert(ACauto* root,char* str)
{
    int len=strlen(str);
    ACauto* now=root;
    for (int i=0;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=0;i<26;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=0,len=strlen(str);
    ACauto* p=root;
    for (int i=0;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>=0)
            {
                ans+=tmp->sum;
                tmp->sum=-1;
            }
            else
            break;
            tmp=tmp->fail;
        }
    }
    return ans;
}

void submain()
{
    ACauto* root=new ACauto;
    int n;
    char str[1000000+50];
    scanf("%d",&n);
    for (int i=0;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=0;kase<T;kase++) submain();
    return 0;
}