【bzoj3172】: [Tjoi2013]单词 字符串-AC自动机

【bzoj3172】: [Tjoi2013]单词

先用所有单词构造一个AC自动机

题目要求的是每个单词在这个AC自动机里匹配到的次数

每次insert一个单词的时候把路径上的cnt++

那么点p->cnt就是以root到p这条路径为前缀的单词的个数

如果p->fail指向了点q,那么就会对q点产生p->cnt的贡献(root到q一定为root到p的后缀)

最后递推统计完所有fail的贡献,找到关键点输出就可以了

 /* http://www.cnblogs.com/karl07/ */
#include <cstdlib>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std; struct trie{
trie *next[],*fail;
int cnt,x;
}t[]; int n;
char s[];
trie *root,*NEW=t;
trie *Q[],*wh[]; trie *new1(int x){NEW++; NEW->cnt=; NEW->x=x; return NEW;} trie *insert(trie *p,int x){
if (!p->next[x]) p->next[x]=new1(x);
p->next[x]->cnt++;
return p->next[x];
} #define pnf p->next[i]->fail
void build_fail(){
int l=,r=;
for (int i=;i<;i++) if (root->next[i]) { Q[++r]=root->next[i]; root->next[i]->fail=root;}
while (l!=r){
trie *p=Q[++l];
for (int i=;i<;i++){
if (p->next[i]){
Q[++r]=p->next[i];
for (pnf=p->fail ; pnf!=root && !pnf->next[i] ; pnf=pnf->fail);
if (pnf->next[i]) pnf=pnf->next[i];
}
}
}
for (int i=r;i>=;i--) Q[i]->fail->cnt+=Q[i]->cnt;
} int main(){
scanf("%d",&n);
root=new1(-);
for (int i=;i<=n;i++){
scanf("%s",s);
wh[i]=root;
int l=strlen(s);
for (int j=;j<l;j++) wh[i]=insert(wh[i],s[j]-'a');
}
build_fail();
for (int i=;i<=n;i++) printf("%d\n",wh[i]->cnt);
return ;
}

一开始zz把strlen放到循环里慢了十几倍

上一篇:linux 运维工程师发展路线


下一篇:iOS 上线被拒收集