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;
}