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$自动机。

$AC$自动机板子题。刚刚复习了一下，基本上还记得，只是$last$数组那里忘了，不过这题也不需要。。

直接构出$trie$树和$fail$树，然后按照$fail$树自底向上更新单词数量就行了。

 

//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <complex>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define N (1000010)
#define il inline
#define RG register
#define ll long long
#define File(s) freopen(s".in","r",stdin),freopen(s".out","w",stdout)

using namespace std;

int q[N],pos[210],n,len;
char s[N];

struct AC_auto{

    int ch[N][26],val[N],nxt[N],last[N],sz;
    
    il void insert(char *s,RG int len,RG int p){
    RG int x=0,c;
    for (RG int i=1;i<=len;++i){
        c=s[i]-97; if (!ch[x][c]) ch[x][c]=++sz;
        x=ch[x][c],val[x]++;
    }
    pos[p]=x; return;
    }
    
    il void build(){
    RG int h=0,t=0;
    for (RG int c=0;c<26;++c)
        if (ch[0][c]) nxt[ch[0][c]]=last[ch[0][c]]=0,q[++t]=ch[0][c];
    while (h<t){
        RG int x=q[++h],v,j;
        for (RG int c=0;c<26;++c){
        v=ch[x][c];
        if (!v){ ch[x][c]=ch[nxt[x]][c]; continue; }
        q[++t]=v,j=nxt[x]; while (j && !ch[j][c]) j=nxt[j];
        nxt[v]=ch[j][c],last[v]=val[nxt[v]] ? nxt[v] : last[nxt[v]];
        }
    }
    for (RG int i=t;i;--i) val[nxt[q[i]]]+=val[q[i]]; return;
    }
    
}AC;

il int gi(){
    RG int x=0,q=1; RG char ch=getchar();
    while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
    if (ch=='-') q=-1,ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
    return q*x;
}

il void work(){
    n=gi();
    for (RG int i=1;i<=n;++i){
    scanf("%s",s+1),len=strlen(s+1);
    AC.insert(s,len,i);
    }
    AC.build();
    for (RG int i=1;i<=n;++i)
    printf("%d\n",AC.val[pos[i]]);
    return;
}

int main(){
    File("word");
    work();
    return 0;
}