【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=;
 const int MAXLEN=;
 int n;
 int cnt=;
 struct ACauto
 {
     ACauto* next[Lnum];
     ACauto* fail;
     int id,sum;
     ACauto()
     {
         for (int i=;i<Lnum;i++) next[i]=NULL;
         fail=NULL;
         sum=;
         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=;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=;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=;i<n;i++)
     {
         scanf("%s",str);
         insert(rt,str,i);
     }
     build(rt);
 }

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

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