AC自动机例题荟萃

洛谷P3808 【模板】AC自动机（简单版），

洛谷P3796 【模板】AC自动机（加强版）

模板体，不会看看  题解  吧

hdu2222 Keywords Search

模板题，呃，放个ac代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=1000010;
const int maxm=500500;
char t[60],s[maxn];
int n;
int ch[maxm][26];//trie树节点
int val[maxm];//每个字符串的结尾都有一个非0的val
int fail[maxm];//前缀指针
int last[maxm];//最长前后缀的结尾位置
int sz;
void insert( char *s)
{
	int u=0;
	int n=strlen(s);
	for(int i=0;i<n;i++)
	{
		int c=s[i]-'a';
		if(!ch[u][c])
		{
			memset(ch[sz],0,sizeof(ch[sz]));
			val[sz]=0;
			ch[u][c]=sz++;
		}
		u=ch[u][c];
	}
	val[u]++;
}
void getfail()
{
	queue<int> q;
	fail[0]=0;
	int u=0;
	for(int i=0;i<26;i++)//先把第一层儿子放进去
	{
		u=ch[0][i];
		if(u)
		{
			q.push(u);
			fail[u]=0;//第一层儿子（个字母）失配后一定回到根节点
			last[u]=0;
		}
	}
	while(!q.empty())
	{
		int r=q.front();q.pop();//把栈顶的取出来
		for(int i=0;i<26;i++)
		{
			u=ch[r][i];//r第i个儿子（下一位字母）的编号
			if(!u)
			{
				ch[r][i]=ch[fail[r]][i];
				continue;
			}
			q.push(u);
			int v=fail[r];
			while(v&&!ch[v][i]) v=fail[v];//找到可以的节点继续
			fail[u]=ch[v][i];
			last[u]=val[fail[u]]? fail[u] : last[fail[u]];
		}
	}
}
int find(char *s)
{
	int u=0,cnt=0;
	int n=strlen(s);
	for(int i=0;i<n;i++)
	{
		int c=s[i]-'a';
		u=ch[u][c];
		int temp=0;
		if(val[u]) temp=u;
		else if(last[u]) temp=last[u];
		while(temp)
		{
			cnt+=val[temp];
			val[temp]=0;
			temp=last[temp];
		}
	}
	return cnt;
}
int main()
{
	freopen("ks.in","r",stdin);
	int T;
	scanf("%d",&T);
	while(T--)
	{
		memset(ch[0],0,sizeof(ch[0]));
		sz=1;
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		{
			scanf("%s",t);
			insert(t);
		}
		getfail();
		scanf("%s",s);
		//printf("%s\n",s);
		int ans=find(s);
		printf("%d\n",ans);
	}
	return 0;
}

洛谷P5231 [JSOI2012]玄武密码

lydsy4327: JSOI2012 玄武密码

#include<bits/stdc++.h>
using namespace std;
const int N=5e5 + 5;
int n,m;
char a[10000005],tmp[N][105];
int ch[N*4][4],fail[N],e[N],tot,c[250];
void clear()
{ 
    memset(fail,0,sizeof(fail));
    memset(e,0,  sizeof(e));
    memset(ch,0, sizeof(ch));
    memset(c,-1,  sizeof(c));
    c['W']=0,c['E']=1,c['S']=2,c['N']=3;
    tot=0;
}
void insert(char *s) {//Trie中的插入，不多解释
    int p=0;
    for(int i=0;s[i];i++)
    {
        int k=c[s[i]];
        if(!ch[p][k]) ch[p][k]=++tot;
        p=ch[p][k];
    }
}
void build()
{
    queue<int> q;
    for(int i=0;i<4;i++)
	{
	 	if(ch[0][i]) q.push(ch[0][i]);
	}
    while(!q.empty())
	{
        int p=q.front();
        q.pop();
        for(int i=0;i<4;i++)
	    {
		   	if(!ch[p][i])
			{
		   		ch[p][i]=ch[fail[p]][i];
		   		continue;
			}
            fail[ch[p][i]]=ch[fail[p]][i];
            q.push(ch[p][i]);
        }
    }
}
void match(char *s)
{
    int p=0;
    for(int i=0;s[i];i++)
	{
        int k=c[s[i]];
        p=ch[p][k];
        for(int j=p;j;j=fail[j])
		{
            if(e[j]) break;
            e[j]=1;
        }
	}
}
int query(char *s)
{
    int j=0,res=0;
    for(int i=0;s[i];i ++)
	{
        int k=c[s[i]];
        j=ch[j][k];
        if(e[j]) res=i + 1;
    }
    return res;
}
int main()
{
	freopen("xuanwu.in","r",stdin);
    scanf("%d%d",&n,&m);
    scanf("%s",a);
    clear();
    for(int i=1;i<=m;i++)
    {
        scanf("%s",tmp[i]);
        insert(tmp[i]);
    }
    build();
	match(a);
    for(int i=1;i<=m;i++) printf("%d\n",query(tmp[i]));
    return 0;
}