【AC自动机】单词

【题目链接】

https://loj.ac/problem/10060

【题意】

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

【题解】

与AC自动机模板题3类似，但是这个题目记录的子串是不能重叠的。

利用下标从后往前历遍，然后把对应的位置的地方叠加到公共后缀。原因是：如果该单词出现过，那么对应的公共后缀肯定也出现过。

#include<cstdio>
#include<algorithm>
using namespace std;
const int N = 1005;
const int M = 1e6+10;
char S[N][205];
int Trie[M][26],fail[M],End[M],Sz[M];
int Q[M],Head,Tail;
int head[M],nxt[M],to[M],cnt;
int n,idx=1;

void Add_edge(int u,int v);
void dfs(int u);
void Insert( char s[] , int Id){
    int p = 1 ;
    for(int i=0;s[i];i++){
        int t = s[i] - 'a' ;
        if( !Trie[p][t] )
            Trie[p][t] = ++idx ;
        p = Trie[p][t] ;
        Sz[p] ++ ;
    }
    End[Id] = p ;
}
void Build(){
    Head = 1 , Tail = 0;
    for(int i=0;i<26;i++) Trie[0][i] = 1 ;

    Q[++Tail] = 1 ;
    while( Head <= Tail ){
        int u = Q[Head++];
        for(int i=0;i<26;i++){
            int To = Trie[u][i];
            if( To ){
                fail[To] = Trie[fail[u]][i];
                Q[++Tail] = To;
            }else{
                Trie[u][i] = Trie[fail[u]][i];
            }
        }
    }
}
void Query(){

    /*for(int i=1,p=1;i<=n;i++){
        p = 1 ;
        for(int j=0;S[i][j];j++){
            int t = S[i][j] - 'a' ;
            p = Trie[p][t];
            Sz[p] ++;
        }
    }*/

    for(int i=idx;i>=1;i--){
//        Add_edge(fail[i],i);
        Sz[fail[Q[i]]] += Sz[Q[i]];
    }
    //dfs(1);
    for(int i=1;i<=n;i++){
        printf("%d
",Sz[End[i]]);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",S[i]);
        Insert(S[i],i);

        //printf("### %s ### 
",S[i]);
    }
    Build();
    //printf("@@@@
");
    Query();
    return 0;
}
void dfs(int u){
    for(int i=head[u];i;i=nxt[i] ){
        int To = to[i] ;
        dfs( To ) ;
        Sz[u] += Sz[To] ;
    }
}
void Add_edge(int u,int v){
    nxt[++cnt] = head[u];
    head[u] = cnt ;
    to[cnt] = v ;
}