BZOJ3172: [Tjoi2013]单词

【传送门：BZOJ3172】

---

简要题意：

　　给出n个单词，你可以理解为将这些单词变成一个个段落，然后求出每个单词在所有段落中出现的次数

---

题解（一）：

　　刚开始不是很懂题目，结果发现将所有单词看成一篇文章，每个单词看成一个段落就懂了

　　由于某种unbelievable的原因，我刚好做了AC自动机的专题训练，看到这道题就秒想AC自动机

　　将每个单词放进AC自动机里，每个点的s表示有多少个单词经过，然后在构建失败指针的时候，通过队列来更新s值，怎么更新呢，假设有一个i点，它的失败指针指向j，设sj为从根到j点所构成的字符串，si为从根到i点所构成的字符串，那么我们可以知道sj为si的后缀，也说明了si中有sj的出现，那么就将j点的s值加上i点的s值，达到更新且不重复的目的来求出答案

---

 

参考代码（一）：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
using namespace std;
struct node
{int s,c[27],fail;node(){s=fail=0;memset(c,-1,sizeof(c));}
}t[1100000];
int tot,ans,n;
char a[1100000];
int ed[210];
void bt(int k,int root)
{int x=root,len=strlen(a+1);for(int i=1;i<=len;i++){int y=a[i]-'a'+1;if(t[x].c[y]==-1){t[x].c[y]=++tot;}x=t[x].c[y];t[x].s++;}ed[k]=x;
}
int list[1100000];
void bfs()
{int x;int head=1,tail=1;list[1]=0;while(head<=tail){x=list[head];for(int i=1;i<=26;i++){int son=t[x].c[i];if(son==-1)continue;if(x==0) t[son].fail=0;else{int j=t[x].fail;while(j!=0&&t[j].c[i]==-1) j=t[j].fail;t[son].fail=max(t[j].c[i],0);int x=t[son].fail,y=son;}list[++tail]=son;}head++;}for(int i=tail;i>=1;i--) t[t[list[i]].fail].s+=t[list[i]].s;
}
int main()
{ans=tot=0;scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%s",a+1);bt(i,0);}bfs();for(int i=1;i<=n;i++){printf("%d\n",t[ed[i]].s);}return 0;
}

 

---

 

题解（二）：

　　  A了这道题之后才发现早在之前就A了（心酸）

　　那我们就来搞一下第二种想法

　　很容易就想到AC自动机，但是单单是AC自动机还不行

　　这时就要用AC自动机的延伸——fail树来做（时常膜一膜算法，有益身体健康）

　　fail树这个玩意就是将AC自动机中fail指针当成是一条边，然后建成一棵树

　　由于trie树上的每个点相当于一个字符串，所以这棵fail树父亲节点是儿子节点所构成字符串的最长后缀

　　这样子对于这道题而言，fail树简直就是神一般的存在QAQ

　　首先将每个字符串放进trie树里面，并且每经过一个trie树里的点，就用s数组记录这个字符串出现的个数，每经过一次就加一，然后构造AC自动机，然后在构造AC自动机的过程中，对每个点的fail指针进行建边，然后用dfs从根往下遍历，把s数组从下往上累加

　　然后在输入每个字符串的时候，记录每个字符串的最后一个字符在trie树上的编号，然后一个一个输出s[end[i]]（end[i]表示第i个字符串最后一个字母在trie树上的编号）

---

参考代码（二）：

#include<queue>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#include<cmath>
using namespace std;
struct node
{int c[27],fail;node(){fail=0;memset(c,-1,sizeof(c));}
}t[1100000];
int tot,n;
char st[1100000];
int s[1100000];
int end[1100000];
void bt(int root,int z)
{int x=root,len=strlen(st+1);for(int i=1;i<=len;i++){int y=st[i]-'a'+1;if(t[x].c[y]==-1){t[x].c[y]=++tot;}x=t[x].c[y];s[x]++;}end[z]=x;
}
struct edge
{int x,y,next;
}a[1100000];int len,last[1100000];
void ins(int x,int y)
{len++;a[len].x=x;a[len].y=y;a[len].next=last[x];last[x]=len;
}
queue<int> q;
void bfs()
{int x;q.push(0);while(q.empty()==0){x=q.front();for(int i=1;i<=26;i++){int son=t[x].c[i];if(son==-1)continue;if(x==0) t[son].fail=0;else{int j=t[x].fail;while(j!=0&&t[j].c[i]==-1) j=t[j].fail;t[son].fail=max(t[j].c[i],0);}ins(t[son].fail,son);q.push(son);}q.pop();}
}
void dfs(int x)
{for(int k=last[x];k;k=a[k].next){int y=a[k].y;dfs(y);s[x]+=s[y];}
}
int main()
{tot=0;scanf("%d",&n);for(int i=1;i<=n;i++){scanf("%s",st+1);bt(0,i);}len=0;memset(last,0,sizeof(last));bfs();dfs(0);for(int i=1;i<=n;i++) printf("%d\n",s[end[i]]);return 0;
}