【AC自动机/fail树】BZOJ3172- [Tjoi2013]单词

【题目大意】

http://www.lydsy.com:808/JudgeOnline/problem.php?id=3172

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

【思路】

第一次写fail树。首先建立AC自动机，对于路径上的每一个点sum++，表示出现的次数。fail指针指向的后缀，如果从fail指针指向的点开始方向建立fail树，其子树的sum之和就等于以它作为后缀的串的总数，相当于它在文章中出现的个数。

*fail树好神奇啊，不过E数组到底应该开多大呢？struct类型的数组到底应该怎么计算空间呢？困惑。

【错误名】

AC自动机的fail指针建立错了…详细见注释。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#define Lnum 26
using namespace std;
const int MAXN=201;
const int MAXLEN=1000001; 
int n;
int cnt=0; 
struct ACauto
{
    ACauto* next[Lnum];
    ACauto* fail;
    int id,sum;
    ACauto()
    {
        for (int i=0;i<Lnum;i++) next[i]=NULL;
        fail=NULL;
        sum=0;
        id=++cnt;
    }
};
ACauto* rt=new ACauto();
ACauto* node[MAXN];
struct edge
{
    ACauto* to;
    edge* next;
    edge()
    {
        to=NULL;
        next=NULL;
    }
};
edge* E[MAXLEN*Lnum];

void addedge(ACauto* u,ACauto* v)
{
    edge* tmp=new edge;
    tmp->to=v;
    tmp->next=E[u->id];
    E[u->id]=tmp;
}

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

void build(ACauto* rt)
{
    queue<ACauto*> que;
    que.push(rt);
    while (!que.empty())
    {
        ACauto* head=que.front();que.pop();
        for (int i=0;i<Lnum;i++)
        {
            if (head->next[i]!=NULL)
            {
                if (head==rt)
                {
                    head->next[i]->fail=rt;
                    addedge(rt,head->next[i]);
                }
                else
                {
                    ACauto* tmp=head->fail;
                    while (tmp!=NULL)
                    {
                        if (tmp->next[i]!=NULL)
                        {
                            head->next[i]->fail=tmp->next[i];
                            //指向最长的后缀，所以是tmp->next[i],而不是tmp 
                            addedge(tmp->next[i],head->next[i]);
                            break;
                        }
                        else tmp=tmp->fail;
                    }
                    if (tmp==NULL) 
                    {
                        head->next[i]->fail=rt;
                        addedge(rt,head->next[i]);
                    }
                }
                que.push(head->next[i]);
            }
        }
    }
}

void dfs(ACauto* rt)
{
    for (edge* i=E[rt->id];i!=NULL;i=i->next)
    {
        dfs(i->to);
        rt->sum+=i->to->sum;
    }
}

void init()
{
    scanf("%d",&n);
    char str[MAXLEN];
    for (int i=0;i<n;i++)
    {
        scanf("%s",str);
        insert(rt,str,i);
    }
    build(rt);
}

void getans()
{
    dfs(rt);
    for (int i=0;i<n;i++) cout<<node[i]->sum<<endl;
}

int main()
{
    init();
    getans();
    return 0;
}