bzoj3172 [Tjoi2013]单词

传送门
Description

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

Input

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

Output

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

Sample Input

3

a

aa

aaa
Sample Output

6

3

1

同样是裸的AC自动机，不过处理有点蛋疼。。。一开始处理的不太好结果导致时间是别人的4倍QAQ

CODE：

#include<cstdio>#include<cstring>struct queue{    int h,t;    int a[1000001];    void clear() {h=1,t=0;}    void push(int n) {a[++t]=n;}    int front() {return a[h];}    void pop() {h++;}    bool empty() {return h>t;}}q;struct AC{    int ch[26];    int fail,cnt;}a[1000001];;char s[1000001];int pos[201];int n,tot;inline void init(){    q.clear();    a[0].fail=-1;}inline void insert(char *s,int d){    int now=0,p,len=strlen(s);    for(int i=0;i<len;i++)    {        p=s[i]-'a';        if(!a[now].ch[p]) a[now].ch[p]=++tot;        now=a[now].ch[p];        a[now].cnt++;    }    pos[d]=now;}inline void makefail(){    q.push(0);    while(!q.empty())    {        int now=q.front();        q.pop();        for(int i=0;i<26;i++)          if(a[now].ch[i])          {            int p=a[now].fail;            while(p!=-1&&!a[p].ch[i]) p=a[p].fail;            if(p==-1) a[a[now].ch[i]].fail=0;            else a[a[now].ch[i]].fail=a[p].ch[i];            q.push(a[now].ch[i]);          }    }    for(int i=q.t;i>=1;i--)      a[a[q.a[i]].fail].cnt+=a[q.a[i]].cnt;}int main(){    scanf("%d",&n);    init();    for(int i=1;i<=n;i++)      scanf("%s",s),insert(s,i);    makefail();    for(int i=1;i<=n;i++)      printf("%d\n",a[pos[i]].cnt);    return 0;}