LOJ10057

AC自动机模板题（~~但我只会WA自动机、TLE自动机，CE自动机……~~）
实际上就是把KMP做到了Trie上。。
#include<bits/stdc++.h>
#define gt() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++)
#define pt(ch) (Top<1000000?St[Top++]=ch:(fwrite(St,1,1000000,stdout),St[(Top=0)++]=ch))
using namespace std;
int Top;static char St[1000000],buf[1000000],*p1=buf,*p2=buf;
const int maxs=(1e6)+5;
int T,n,Ans,que[maxs];
struct Trie{
    int tot,c[maxs][27],cnt[maxs],nxt[maxs];
    int clear(int i){return cnt[i]=nxt[i]=0,memset(c[i],0,sizeof c[i]),i;}
    void insert(int x){
        int rot=0;char ch=gt();while(!islower(ch)) ch=gt();
        while(islower(ch)) ch-='a',rot=(c[rot][ch]?c[rot][ch]:c[rot][ch]=clear(++tot)),ch=gt();
        cnt[rot]++;
    }
}AC;
int read(){
    int ret=0;char ch=gt();
    while(ch<'0'||ch>'9') ch=gt();
    while(ch>='0'&&ch<='9') ret=ret*10+ch-'0',ch=gt();
    return ret;
}
void write(int x){if(x>9) write(x/10);pt(x%10+'0');}
void getnxt(){
    int hed=0,tal=0,x;
    for(int i=0;i<26;i++) if(AC.c[0][i]) que[++tal]=AC.c[0][i];
    while(hed!=tal){
        x=que[++hed];
        for(int i=0;i<26;i++)
          if(AC.c[x][i]) AC.nxt[que[++tal]=AC.c[x][i]]=AC.c[AC.nxt[x]][i];
            else AC.c[x][i]=AC.c[AC.nxt[x]][i];
    }
}
int main(){
    T=read();
    while(T--){
        n=read(),Ans=0,AC.clear(AC.tot=0);
        for(int i=1;i<=n;i++) AC.insert(i);
        getnxt();
        int rot=0;char ch=gt();while(!islower(ch)) ch=gt();
        while(islower(ch)){
            rot=AC.c[rot][ch-='a'];
            for(int t=rot;~AC.cnt[t]&&t;t=AC.nxt[t]) Ans+=AC.cnt[t],AC.cnt[t]=-1;
            ch=gt();
        }
        write(Ans),pt('\n');
    }
    fwrite(St,1,Top,stdout);
    return 0;
}
猜你喜欢
