1. 程式人生 > 實用技巧 >P3808 【模板】AC自動機(簡單版)

P3808 【模板】AC自動機(簡單版)

Miku

簡單版就是個單純的模板

ac自動機是啥,就是一個加了類似於kmp的next陣列的tire樹

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<queue>
#include<algorithm>
using namespace std;
struct t{
	int fail;
	int nex[26];
	int end;
}ac[1000001];
int cnt;
int l;
int p;
int x;
int ans;
int n;
string s;
queue <int>q;
void build(string s){
	l=s.length();
	p=0;
	for(int i=0;i<l;++i){
		x=s[i]-'a';
		if(!ac[p].nex[x]){
			cnt++;
			ac[p].nex[x]=cnt;
		}
		p=ac[p].nex[x];
	}
	ac[p].end+=1;//這就是個tire樹 
}
void buildfail(){//找指標要 bfs 
	ac[0].fail=0;//0指自己 
	for(int i=0;i<26;++i){
		if(ac[0].nex[i]!=0){
			ac[ac[0].nex[i]].fail=0;
			q.push(ac[0].nex[i]);
		}//第二層也是 
	}
	while(!q.empty()){
		x=q.front();
		q.pop();
		for(int i=0;i<26;++i){
			if(ac[x].nex[i]!=0){
				ac[ac[x].nex[i]].fail=ac[ac[x].fail].nex[i];//失配指標尋找
				//為它失配指標指向的點的對應字母 
				q.push(ac[x].nex[i]);//改了就壓,繼續更新 
			}else{
				ac[x].nex[i]=ac[ac[x].fail].nex[i];//如果這個點是不存在的
				//那麼直接指向失配指標對應點的對應點 
			}
		}	
	}
}
int query(string s){
	l=s.length();
	p=0;
	ans=0;
	for(int i=0;i<l;++i){
		x=s[i]-'a';
		p=ac[p].nex[x];
		for(int j=p;j&&ac[j].end!=-1;j=ac[j].fail){//跳躍,收集,清空 
			ans+=ac[j].end;
			ac[j].end=-1;
		}
	}
	return ans;
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;++i){
		cin>>s;
		build(s);
	}
	buildfail();
	cin>>s;
	cout<<query(s)<<endl;
	return 0;
}