[FROM LUOGU]【模板】AC自動機(簡單版)

傳送門

【模板複習】

SOL
AC自動機其實就是利用KMP算法的線性性,得到的可以支持多模式串匹配多母串的算法,可以簡單理解爲Trie+KMP,fail指針的意義其實與KMP的nxt數組相似,就是前綴=後綴
算法流程:
1.用模式串構建Trie樹
2.在Trie樹構造fail(失配指針)鏈,如果該節點不存在就將其指向其父親的fail的相應兒子,否則其fail值爲其父親的fail的相應兒子
3.查詢答案

代碼:

#include<bits/stdc++.h>
using namespace std;
#define re register
const int N=1e6+5,SIZ=26;
int n,cnt,ch[N][SIZ],ans,val[N],fail[N];
inline int idx(char c){return c-'a';}
inline void insert(char *t){
	int re u=0,len=strlen(t);
	for(int re i=0;i^len;++i){
		int re c=idx(t[i]);
		if(!ch[u][c])ch[u][c]=++cnt;
		u=ch[u][c];
	}++val[u];
}
inline void getfail(){
	queue<int>q;
	for(int re i=0;i^SIZ;++i)if(ch[0][i])q.push(ch[0][i]);
	while(!q.empty()){
		int re u=q.front();q.pop();
		for(int re i=0;i^SIZ;++i){
			if(!ch[u][i])ch[u][i]=ch[fail[u]][i];
			else fail[ch[u][i]]=ch[fail[u]][i],q.push(ch[u][i]);
		}
	}
}
inline void find(char *t){
	int re u=0,len=strlen(t);
	for(int re i=0;i^len;++i){
		int re c=idx(t[i]);
		for(int re j=ch[u][c];j&&val[j];j=fail[j])ans+=val[j],val[j]=0;
		u=ch[u][c];
	}
}
char s[N];
signed main(){
	scanf("%d",&n);
	for(int re i=1;i<=n;++i)scanf("%s",s),insert(s);
	getfail();
	scanf("%s",s),find(s),cout<<ans,exit(0);
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章