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