BZOJ 3172 [Tjoi2013]单词 AC自动机 模板题

Description

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

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

HINT

Source

题目传送门

开心~又在bzoj找到一题水题。。

注意一下题目题意有点点不清晰。。。

其实总的串是N个单词全部连接起来，而且每两个之间有一个空格（也可以用其它字符代替）

还有就是是N个单词全部连起来的长度<=10^6……

题目比较良心的地方是内存开了512M，这样子的话可以直接上trie[][]了。

总之是一道水水的模板题吧。。

感觉这题加快速读入就是智障

#include<bits/stdc++.h>using namespace std;const intN=1000205,NUM=205;int n,cnt,LenAll;int pp[N];int Ans[NUM],fail[N],num[N],Q[N];int trie[N][27];char All[N],tmp[N];int find(int x,int t){if (trie[x][t]) return trie[x][t];if (!x) return 0;return find(fail[x],t);}void insert(int ii){int len=strlen(tmp),now=0;for (int i=0;i<len;i++){int j=tmp[i]-'a';if (trie[now][j]) now=trie[now][j];else now=trie[now][j]=++cnt;}if (num[now]) pp[ii]=num[now];else num[now]=ii;}void AC(){int head=0,tail=1;Q[0]=0;while (head!=tail){int now=Q[head++];for (int i=0;i<27;i++){if (!trie[now][i]) continue;int t=find(fail[now],i);if (!now) t=0;fail[trie[now][i]]=t;Q[tail++]=trie[now][i];}}}void solve(){int now=0;for (int i=0;i<LenAll;i++){int j=All[i]-'a';now=find(now,j);for (int j=now;j;j=fail[j])if (num[j]) Ans[num[j]]++;}}int main(){    scanf("%d",&n);LenAll=cnt=0;for (int i=1;i<=n;i++){scanf("%s",tmp);int len=strlen(tmp);for (int j=0;j<len;j++) All[LenAll++]=tmp[j];All[LenAll++]='{';insert(i);}AC();solve();for (int i=1;i<=n;i++)        if (pp[i]) Ans[i]=Ans[pp[i]];for (int i=1;i<=n;i++)printf("%d\n",Ans[i]);return 0;}