BZOJ3172单词

3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 2536 Solved: 1185
Description
某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。
Input
第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6
Output
输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。
Sample Input
3
a
aa
aaa
Sample Output
6
3
1
借鉴的黄学长的模板。。
建出fail树之后。。对每个子串的每个节点打标记。。
求出每个子串的终点标记个数和。。
附上本蒟蒻的代码：

#include<cstdio>
#include<cstring>
using namespace std;
#define MAXN 1000001
int n,pos[MAXN];
struct kx
{
    int cnt,next[MAXN][26],sum[MAXN],fail[MAXN],q[MAXN];
    char ch[MAXN];
    kx()
      {
        cnt=1;
        for (int i=0;i<26;i++) next[0][i]=1;
      }
    void insert(int &pos)
      {
        int now=1;
        scanf("%s",&ch);
        int n=strlen(ch);
        for (int i=0;i<n;i++)
          {
            if (!next[now][ch[i]-'a']) next[now][ch[i]-'a']=++cnt;
            now=next[now][ch[i]-'a'];
            sum[now]++;
          }
        pos=now;
      }
    void build()
      {
        int head=0,tail=1;
        q[0]=1,fail[1]=0;
        while (head!=tail)
          {
            int now=q[head];head++;
            for (int i=0;i<26;i++)
              {
                int v=next[now][i];
                if (!v) continue;
                int k=fail[now];
                while (!next[k][i]) k=fail[k];
                fail[v]=next[k][i];
                q[tail++]=v;
              }
          }
        for (int i=tail-1;i>=0;i--)
          sum[fail[q[i]]]+=sum[q[i]];
      }
}kx;

int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
      kx.insert(pos[i]);
    kx.build();
    for (int i=1;i<=n;i++)
      printf("%d\n",kx.sum[pos[i]]);
    return 0;
}