模板 AC自动机

模板 AC自动机

code：

#include <iostream>
#include <cstdio>
#include <queue>
#include <cstring>

using namespace std;

const int wx=2000017;

inline int read(){
	int sum=0,f=1; char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=(sum<<1)+(sum<<3)+ch-'0'; ch=getchar();}
	return sum*f;
}

struct AC_automation{
	int tr[wx][26];
	int val[wx],fail[wx];
	int cnt;
	
	void insert(char * s){
		int p=0;
		for(int i=0;s[i];i++){
			int k=s[i]-'a';
			if(!tr[p][k])tr[p][k]=++cnt;
			p=tr[p][k];
		}
		val[p]++;
	}
	
	void build(){
		queue<int > q; memset(fail,0,sizeof fail);
		for(int i=0;i<26;i++)if(tr[0][i])q.push(tr[0][i]);
		while(q.size()){
			int p=q.front(); q.pop();
			for(int i=0;i<26;i++){
				if(tr[p][i]){
					fail[tr[p][i]]=tr[fail[p]][i];
					q.push(tr[p][i]);
				}
				else tr[p][i]=tr[fail[p]][i];
			}
		}
	}
	
	int query(char * s){
		int p=0,re=0;
		for(int i=0;s[i];i++){
			p=tr[p][s[i]-'a'];
			for(int j=p;j&&~val[j];j=fail[j])if(val[j]!=-1)re+=val[j],val[j]=-1;
		}
		return re;
	}
	
}ac;

int n;
char c[wx];

int main(){
	n=read();
	for(int i=1;i<=n;i++){
		scanf("%s",c);
		ac.insert(c);
	}
	ac.build();
	scanf("%s",c);
	printf("%d\n",ac.query(c));
	return 0;
}