BZOJ3172 单词
题解
裸的AC自动机，构造出AC自动机之后，将每一个节点的出现次数都累加到它的(fail)节点上去，最后得到的就是答案了。
ｃｏｄｅ
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
bool Finish_read;
template<class T>inline void read(T &x){Finish_read=0;x=0;int f=1;char ch=getchar();while(!isdigit(ch)){if(ch=='-')f=-1;if(ch==EOF)return;ch=getchar();}while(isdigit(ch))x=x*10+ch-'0',ch=getchar();x*=f;Finish_read=1;}
template<class T>inline void print(T x){if(x/10!=0)print(x/10);putchar(x%10+'0');}
template<class T>inline void writeln(T x){if(x<0)putchar('-');x=abs(x);print(x);putchar('
');}
template<class T>inline void write(T x){if(x<0)putchar('-');x=abs(x);print(x);}
/*================Header Template==============*/
const int maxn=1e6+500;
int n,w[210];
char s[maxn];
/*==================Define Area================*/
struct ac_automaton{
int tot,ch[maxn][26],fail[maxn],q[maxn],head,tail,sum[maxn];
void insert(int &x){
int p=0;
int len=strlen(s);
for (int i=0;i<len;i++){
if (!ch[p][s[i]-'a']) ch[p][s[i]-'a']=++tot;
++sum[p=ch[p][s[i]-'a']];
}
x=p;
}
void getfail(){
head=0,q[tail=1],fail[0]=-1;
while (head!=tail){
int x=q[++head];
for (int i=0;i<26;i++){
if (ch[x][i]) q[++tail]=ch[x][i],fail[ch[x][i]]=x==0?0:ch[fail[x]][i];
else ch[x][i]=x==0?0:ch[fail[x]][i];
}
}
}
void work(){
for (int i=tail;i;i--) sum[fail[q[i]]]+=sum[q[i]];
for (int i=1;i<=n;i++) printf("%d
",sum[w[i]]);
}
}T;
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%s",s),T.insert(w[i]);
T.getfail(),T.work();
return 0;
}