BZOJ_3172_[Tjoi2013]单词_AC自动机
Description
某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
Input
第一个一个整数N,表示有多少个单词,接下来N行每行一个单词。每个单词由小写字母组成,N<=200,单词长度不超过10^6
Output
输出N个整数,第i行的数字表示第i个单词在文章中出现了多少次。
Sample Input
3
a
aa
aaa
a
aa
aaa
Sample Output
6
3
1
3
1
对所有串建立trie图,插入时对沿途所有经过的点标记++。
然后在fail树上求每个子串对应位置的子树标记和即可。
由于bfs拓展是按层拓展,所以这里直接倒着扫一遍队列。
代码:
#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 1000050
int ch[N][26],cnt=1,fail[N],flg[N],n,Q[N],l,r,pos[N];
char w[N];
void insert(int x) {
int i,p=1;
for(i=1;w[i];i++) {
int &k=ch[p][w[i]-'a'];
if(!k) k=++cnt;
p=k;
flg[p]++;
}
pos[x]=p;
}
void build() {
int i;
for(i=0;i<26;i++) ch[0][i]=1;
Q[r++]=1;
while(l<r) {
int p=Q[l++];
for(i=0;i<26;i++) {
if(ch[p][i]) fail[ch[p][i]]=ch[fail[p]][i],Q[r++]=ch[p][i];
else ch[p][i]=ch[fail[p]][i];
}
}
}
int main() {
scanf("%d",&n);
int i;
for(i=1;i<=n;i++) {
scanf("%s",w+1);
insert(i);
}
build();
for(i=cnt;i;i--) flg[fail[Q[i]]]+=flg[Q[i]];
for(i=1;i<=n;i++) {
printf("%d\n",flg[pos[i]]);
}
}