[TJOI2013]单词

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

直接构造trie和AC自动机(fail树)

然后把文章拼出来，然后在AC自动机上，分别统计每个单词的出现次数

但这个方法很慢

实际上可以用前缀和思想，从fail树从下往上累加，省去了最慢的查询

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
const int N=1e6+205;
char s[N];
int Q[N];
int val[N],ch[N][28],size,f[N],n,mp[205],cnt[N];
void insert(int len,int id)
{int i;
  int now=0;
  for (i=0;i<len;i++)
    {
      if (ch[now][s[i]-'a']==0) ch[now][s[i]-'a']=++size;
      now=ch[now][s[i]-'a'];
      cnt[now]++;
    }
  mp[id]=now;
}
void AC_build()
{int i,h,t;
  h=0;t=0;
  for (i=0;i<26;i++)
    if (ch[0][i]) f[ch[0][i]]=0,Q[++t]=ch[0][i];
  while (h<t)
    {
      h++;
      int u=Q[h];
      for (i=0;i<26;i++)
    {
      if (ch[u][i])
        {
          f[ch[u][i]]=ch[f[u]][i];Q[++t]=ch[u][i];
        }
      else ch[u][i]=ch[f[u]][i];
    }
    }
  for (i=t;i>=1;i--)
    cnt[f[Q[i]]]+=cnt[Q[i]];
}
int main()
{int i,len,j;
  cin>>n;
  for (i=1;i<=n;i++)
    {
      scanf("%s",s);
      len=strlen(s);
      insert(len,i);
    }
  AC_build();
  for (i=1;i<=n;i++)
    printf("%d
",cnt[mp[i]]);
}

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
queue<int>Q;
const int N=1e6+205;
char T[N],s[N];
int ans[205],val[N],ch[N][28],size,f[N],cnt,n,mp[205];
void insert(int len,int id)
{int i;
  int now=0;
  for (i=0;i<len;i++)
    {
      if (ch[now][s[i]-'a']==0) ch[now][s[i]-'a']=++size;
      now=ch[now][s[i]-'a'];
    }
  if (val[now]==0)
    val[now]=id,mp[id]=id;
  else mp[id]=val[now];
}
void AC_build()
{int i;
  for (i=0;i<27;i++)
    if (ch[0][i]) f[ch[0][i]]=0,Q.push(ch[0][i]);
  while (Q.empty()==0)
    {
      int u=Q.front();
      Q.pop();
      for (i=0;i<27;i++)
    {
      if (ch[u][i]) f[ch[u][i]]=ch[f[u]][i],Q.push(ch[u][i]);
      else ch[u][i]=ch[f[u]][i];
    }
    }
}
void query()
{int i,j;
  int now=0,len=strlen(T);
  for (i=0;i<len;i++)
    {
      now=ch[now][T[i]-'a'];
      for (j=now;j;j=f[j])
    if (val[j])
      ans[val[j]]++;
      //cout<<i<<' '<<val[j]<<endl;
    }
}
int main()
{int i,len,j;
  cin>>n;
  cnt=-1;
  for (i=1;i<=n;i++)
    {
      scanf("%s",s);
      len=strlen(s);
      insert(len,i);
      T[++cnt]='a'+26;
      for (j=0;j<len;j++)
    T[++cnt]=s[j];
      //T[++cnt]='a'+26;
    }
  AC_build();
  query();
  for (i=1;i<=n;i++)
    printf("%d
",ans[mp[i]]);
}