Ac自动机练习

 

Ac自动机是用来解决

多个小串（模式串），在一个大串（文本串）中出现次数这种类似问题的

 

因为Trie树上直接跑dfs时间复杂度太大，所以考虑减少复杂度

fail指针，相当于每次存了个扫到的当前最长后缀

Trie树上用bfs建fail指针。暴力扫文本串即可

 

 

洛谷p3808，简单模板

给定n个模式串和1个文本串，求有多少个模式串在文本串里出现过。

/*
    author:revolIA
    submit:;
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=1e6+7;
int Trie[maxn][26];
int val[maxn];
int fail[maxn];
int tot;
char s[maxn];
void Insert(char *p,int pos = 0){
    for(;*p;p++){
        int to = *p-'a';
        if(!Trie[pos][to]){
            Trie[pos][to] = ++tot;
        }
        pos = Trie[pos][to];
    }
    val[pos]++;
}
void bfs(){
    queue<int> Q;
    for(int i=0;i<26;i++)if(Trie[0][i]){
        fail[Trie[0][i]] = 0;
        Q.push(Trie[0][i]);
    }
    while(!Q.empty()){
        int now = Q.front();
        Q.pop();
        for(int i=0;i<26;i++){
            if(Trie[now][i]){
                fail[Trie[now][i]] = Trie[fail[now]][i];
                Q.push(Trie[now][i]);
            }else
                Trie[now][i] = Trie[fail[now]][i];
        }
    }
}
int query(char *p,int pos = 0,int ans = 0){
    for(;*p;p++){
        int to = *p-'a';
        pos = Trie[pos][to];
        for(int j=pos;j && ~val[j];j=fail[j]){
            ans += val[j];
            val[j] = -1;
        }
    }
    return ans;
}
int main(){
    int n;
    scanf("%d",&n);
    while(n--)scanf("%s",s),Insert(s);
    bfs();
    scanf("%s",s);
    printf("%d\n",query(s));
    return 0;
}

 

 

 

洛谷p3796 加强模板

有N个由小写字母组成的模式串以及一个文本串T。每个模式串可能会在文本串中出现多次。你需要找出哪些模式串在文本串T中出现的次数最多。

/*
    author:revolIA
    submit:;
*/
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+7;
int Trie[maxn][26];
int val[maxn],idx[maxn];
int fail[maxn];
int ans[155];
int tot;
char T[maxn<<3],s[155][75];
void Insert(char *p,int i = 0,int pos = 0){
    for(;*p;p++){
        int to = *p-'a';
        if(!Trie[pos][to]){
            Trie[pos][to] = ++tot;
            for(int i=0;i<26;i++)Trie[tot][i] = 0;
        }
        pos = Trie[pos][to];
    }
    val[pos]++;
    idx[pos] = i;
}
void bfs(){
    queue<int> Q;
    for(int i=0;i<26;i++)if(Trie[0][i]){
        fail[Trie[0][i]] = 0;
        Q.push(Trie[0][i]);
    }
    while(!Q.empty()){
        int now = Q.front();
        Q.pop();
        for(int i=0;i<26;i++){
            if(Trie[now][i]){
                fail[Trie[now][i]] = Trie[fail[now]][i];
                Q.push(Trie[now][i]);
            }else
                Trie[now][i] = Trie[fail[now]][i];
        }
    }
}
void query(char *p,int pos = 0){
    for(;*p;p++){
        int to = *p-'a';
        pos = Trie[pos][to];
        for(int j=pos;j;j=fail[j]){
            ans[idx[j]]++;
        }
    }
}
int main(){
    int n;
    while(~scanf("%d",&n)&&n){
        tot = 0;
        for(int i=0;i<26;i++)Trie[0][i] = 0;
        memset(val,0,sizeof val);
        memset(idx,0,sizeof idx);
        memset(fail,0,sizeof fail);
        memset(ans,0,sizeof ans);
        for(int i=1;i<=n;i++)scanf("%s",s[i]),Insert(s[i],i);
        bfs();
        scanf("%s",T);
        query(T);
        int cnt = 0;
        for(int i=1;i<=n;i++)cnt = max(cnt,ans[i]);
        printf("%d\n",cnt);
        for(int i=1;i<=n;i++)if(cnt == ans[i])printf("%s\n",s[i]);
    }
    return 0;
}