AC自动机——多个kmp匹配

（并不能自动AC）

介绍：

Aho-Corasick automaton，最经典的处理多个模式串的匹配问题。

是kmp和字典树的结合。

 

精髓与灵魂：

①利用trie处理多个模式串

②引入fail指针。节点x的fail表示，trie中最大的某个前缀等于x到根节点字符串后缀的节点位置。

fail类比于kmp的nxt数组，可以在失配的时候，O（1）找到最大的可能能继续匹配的位置。

所以，ac自动机可看做多个kmp

 

 

步骤：（完整代码在下面）

①建trie树。插入模式串。

void ins(char *s){
        int len=strlen(s+1);
        int now=0;
        for(int i=1;i<=len;i++){
            int x=s[i]-'A';
            if(!a[now][x]) a[now][x]=++cnt;
            now=a[now][x];
        }
        exi[now]=1;
    }

②trie上建ac自动机。

void build(){
        queue<int>q;
        for(int i=0;i<26;i++){
            if(a[0][i]) fail[a[0][i]]=0,q.push(a[0][i]);
        }
        while(!q.empty()){
            int x=q.front();q.pop();
            exi[x]|=exi[fail[x]];
            for(int i=0;i<26;i++){
                if(a[x][i]){
                    fail[a[x][i]]=a[fail[x]][i];
                    q.push(a[x][i]);
                }
                else{
                    a[x][i]=a[fail[x]][i];    
                }
            }
        }
    }

用bfs来建造，并且，即时转移fail指针。

fail指针的转移正确性：

因为bfs是分层加入元素，而fail至少让字符串长度-1，所以之前的fail[x]的各种信息都处理完毕了。

并且，由于fail的定义，所以能在fail[x][i]往下转移，一定就是最优的。

 

这里有个小优化：

else{a[x][i]=a[fail[x]][i];}

如果x没有i这个儿子，那么就直接指向它fail指针位置的儿子。

这样子，在之后的if(a[x][i])中，可以直接走一次fail[x]的儿子就可以找到真正的fail[a[x][i]]了。

因为，如果fail没有这个儿子，不加这个优化还要继续跳fail，复杂度没有保证了。

这样，就类似于并查集的路径压缩思想，直接指到最长的有这个儿子的点了。

（语言表达不好，自行画图理解吧。。。）

 

完整代码：

struct node{
	int a[N*N][26],cnt;
	int fail[N*N];
	bool exi[N*N];
	void init(){
		memset(a,0,sizeof a);memset(exi,0,sizeof exi);
		cnt=0;memset(fail,0,sizeof fail);
	}
	void ins(char *s){
		int len=strlen(s+1);
		int now=0;
		for(int i=1;i<=len;i++){
			int x=s[i]-'A';
			if(!a[now][x]) a[now][x]=++cnt;
			now=a[now][x];
		}
		exi[now]=1;
	}
	void build(){
		queue<int>q;
		for(int i=0;i<26;i++){
			if(a[0][i]) fail[a[0][i]]=0,q.push(a[0][i]);
		}
		while(!q.empty()){
			int x=q.front();q.pop();
			exi[x]|=exi[fail[x]];
			for(int i=0;i<26;i++){
				if(a[x][i]){
					fail[a[x][i]]=a[fail[x]][i];
					q.push(a[x][i]);
				}
				else{
					a[x][i]=a[fail[x]][i];	
				}
			}
		}
	}
}ac;

 

另外，我们ac自动机上节点上，也可以加上其他的标记。

 

例题：

[JSOI2007]文本生成器

Description：

给n个模式串，求有多少个长度为m的文章，至少包含一个模式串

Solution：

Ac自动机的标志很明显，多个模式串，一个主串。

Ac自动机dp的状态很套路，一般就是匹配到j位置，怎么怎么样。。

设f[i][j]，前i个字符，匹配到AC自动机的j位置，没出现一个模式串的方案数。（最后总方案-没出现一个方案，差分）

每个点有一个exi布尔数组，表示匹配到这个点，这个点所代表的前缀（可以是一个完整模式串）是否已经包含了至少一个模式串。

插入的时候，末尾exi=1，bfs的时候，exi[x]|=exi[fail[x]]即可。正确性同bfs分层图性质。

然后判断，直接转移就好了。

代码：

#include<bits/stdc++.h>
using namespace std;
const int N=100+3;
const int mod=10007;
struct node{
    int a[N*N][26],cnt;
    int fail[N*N];
    bool exi[N*N];
    void init(){
        memset(a,0,sizeof a);memset(exi,0,sizeof exi);
        cnt=0;memset(fail,0,sizeof fail);
    }
    void ins(char *s){
        int len=strlen(s+1);
        int now=0;
        for(int i=1;i<=len;i++){
            int x=s[i]-'A';
            if(!a[now][x]) a[now][x]=++cnt;
            now=a[now][x];
        }
        exi[now]=1;
    }
    void build(){
        queue<int>q;
        for(int i=0;i<26;i++){
            if(a[0][i]) fail[a[0][i]]=0,q.push(a[0][i]);
        }
        while(!q.empty()){
            int x=q.front();q.pop();
            exi[x]|=exi[fail[x]];
            for(int i=0;i<26;i++){
                if(a[x][i]){
                    fail[a[x][i]]=a[fail[x]][i];
                    q.push(a[x][i]);
                }
                else{
                    a[x][i]=a[fail[x]][i];    
                }
            }
        }
    }
}ac;
int n,m;
int f[N][N*N];
char s[N];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%s",s+1);
        ac.ins(s);
    }
    ac.build();
    int tot=ac.cnt;
    f[0][0]=1;
    for(int i=0;i<=m-1;i++){
        for(int j=0;j<=tot;j++){
            if(!ac.exi[j]){
                for(int k=0;k<26;k++){
                    if(!ac.exi[ac.a[j][k]]){
                        f[i+1][ac.a[j][k]]=(f[i+1][ac.a[j][k]]+f[i][j])%mod;
                    }
                }
            }
        }
    }
    long long ans=1;
    for(int i=1;i<=m;i++){
        ans=(ans*26)%mod;
    }
    for(int i=0;i<=tot;i++){
        if(!ac.exi[i]){
            ans=(ans-f[m][i]+mod)%mod;
        }
    }
    printf("%lld",ans);
    return 0;
}

 

为什么要记录AC自动机上匹配到的状态呢？

因为，单纯记录这一位是哪个字符肯定不能判断是否包含。

而AC自动机本身的fail，就蕴含了所有的可能包含的位置。只要不断跳fail即可。

相当于该状态已经包罗万象。

因为fail的定义，也不会包含更多，不会包含更少。

所以对于匹配问题再适合不过了。

 

AC自动机dp出题人出烦了之后，

就开始出一些涉及AC自动机形态的题目，更贴近算法本身。

基本开刀处都是fail指针（AC自动机的精髓嘛）

 

[POI2000]病毒

这是一个利用fail指针构可能可以和trie树构成环的特性。从而构造出无限长的串。

[NOI2011]阿狸的打字机——AC自动机之fail树的利用

这个题目恰好相反，把fail树和trie树都利用起来，并且离线处理。

 

这两个题目都是值得思考总结的。都利用了fail的性质，但是都没有直接使用fail指针。妙哉！

 

转载于:https://www.cnblogs.com/Miracevin/p/9416253.html