题目描述:戳这里
题解:
这题是AC自动机的模板题。
AC自动机可以看作是Trie和kmp的结合体(我连kmp都不会写了。。。)
它可以看作是kmp的延伸。kmp求的是某一字符串在给定字符串中出现的次数,而AC自动机求的是某一些字符串在给定字符串中出现的次数之和或者这一类的问题。
AC自动机是用Trie来实现的,只是多了一个fail(指针)。
那么我们就重点来介绍fail的作用。
我们假设有这样一颗trie树,那么每一个点的fail就是图中红色的点。
我们考虑这些fail指针是怎样构造的。
fail[x]=fail[fa[x]].son。
意思就是这个点的fail指针指向的是它的父节点的fail指针指向的点的子节点。
这样的构造方法和kmp就有点类似了。
这样构造的话,我们当前枚举原串(文章)S中的第i个节点S[i]如果和trie树中当前节点不匹配,就可以通过虚边(也就是在构造的时候将某一个点连下去不存在的节点向bfs序最大的并且末尾是这个节点的点连边)直接跳到另一颗子树中去。如果存在的话,也可以通过fail遍历所有的可行叶节点。
这样就基本上讲完了构造和查询,具体可以见代码(因为被大佬模改,码风有点混乱。。。):
#include<cstdio>
#include<string>
#include<cstring>
using namespace std;
const int maxn=1e6+5;
int n,tot,que[maxn];
char s[maxn],s1[maxn];
struct dyt{
int fail,nxt[28],end;
}tree[maxn];
void tle(){while (1);}
void insert(){//建trie树
int len=strlen(s1),now=0;
for (int i=0;i<len;i++){
int x = s1[i] - 'a' + 1;
if (tree[now].nxt[x] == 0) tree[now].nxt[x]=++tot;
now = tree[now].nxt[x];
}
tree[now].end++;
}
void build(){//构造fail
int head=0,tail=0;
for (int i=1;i<=26;i++)
if (tree[0].nxt[i]>0){
tree[tree[0].nxt[i]].fail=0;
que[++tail]=tree[0].nxt[i];
}
while (head!=tail) {
head++; int x=que[head];
for (int i=1;i<=26;i++)
if (tree[x].nxt[i]>0) {
tree[tree[x].nxt[i]].fail = tree[tree[x].fail].nxt[i];
que[++tail]=tree[x].nxt[i];
} else tree[x].nxt[i]=tree[tree[x].fail].nxt[i];
}
}
int get(){//查询
int len=strlen(s); int now=0,ans=0;
for (int i=0;i<len;i++) {
now = tree[now].nxt[s[i]-'a'+1];
for (int j=now;j && tree[j].end > 0 ;j = tree[j].fail){
ans += tree[j].end; tree[j].end=-1;
}
}
return ans;
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++) {
scanf("%s",s1);
insert();
}
build();
scanf("%s",s);
printf("%d\n",get());
return 0;
}