AC自动机模板题

AC自动机学习博客

　　AC自动机理解要点：

　　　　1）fail指针指向的是每个节点，在字典树上和这个节点后缀相同的最长单词，每次都这样匹配，必定不会漏过答案。

　　　　2）字典树建立后，会在bfs求fail阶段把字典树变成一个字典树图（不知道理解的对不对），就是把字典树的末尾节点再往下添加一层，并且连接到fail指针指向的相同位子的儿子。

　　　两道模板题和AC代码。

P3808

#include<bits/stdc++.h>
#define clr(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=;
char s[maxn],p[maxn];
int trie[maxn][],cntword[maxn],fail[maxn],cnt=;
int n;
void insert(char *s){
    int root=;
    int si=strlen(s);
    for(int i=;i<si;i++)
    {
        int Next=s[i]-'a';
        if(!trie[root][Next])trie[root][Next]=++cnt;
        root=trie[root][Next];
    }
    cntword[root]++;
}
void getfail(){
    queue<int >q;
    for(int i=;i<;i++)
    {
        if(trie[][i]){
            q.push(trie[][i]);
        }
    }
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int i=;i<;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 *s){
    int now=,ans=;
    int si=strlen(s);
    for(int i=;i<si;i++)
    {
        now=trie[now][s[i]-'a'];
        for(int j=now;j&&cntword[j]!=-;j=fail[j])
        {
            ans+=cntword[j];
            cntword[j]=-;
        }
    }
    return ans;
}
int main(){
    cin>>n;
    while(n--)
    {
        scanf("%s",p);
        insert(p);
    }
    fail[]=;
    getfail();
    scanf("%s",s);
    cout<<query(s)<<endl;
}

P3796

#include<bits/stdc++.h>
#define clr(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int maxn=;
char s[maxn],p[maxn];
int trie[maxn][],cntword[maxn],fail[maxn],cnt=;
int n;
int num[maxn];
map<int,string>dio;
void insert(char *s){
    int root=;
    int si=strlen(s);
    for(int i=;i<si;i++)
    {
        int Next=s[i]-'a';
        if(!trie[root][Next])trie[root][Next]=++cnt;
        root=trie[root][Next];
    }
    dio[root]=s;
    cntword[root]++;
}
void getfail(){
    queue<int >q;
    for(int i=;i<;i++)
    {
        if(trie[][i]){
            q.push(trie[][i]);
        }
    }
    while(!q.empty())
    {
        int now=q.front();
        q.pop();
        for(int i=;i<;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 *s){
    int now=,ans=;
    int si=strlen(s);
    for(int i=;i<si;i++)
    {
        now=trie[now][s[i]-'a'];
        for(int j=now;j&&cntword[j]!=-;j=fail[j])
        {
            //ans+=cntword[j];
            if(cntword[j]>){
                num[j]+=cntword[j];
            }
            //cntword[j]=-1;
        }
    }
}
void init(){
    clr(trie,);
    clr(num,),clr(cntword,);
    dio.clear();
}
int main(){
    while(scanf("%d",&n),n){
        init();
        for(int i=;i<=n;i++)
        {
            scanf("%s",p);
            insert(p);
        }
        int p,maxx=;
        vector<int >ans;
        fail[]=;
        getfail();
        scanf("%s",s);
        query(s);
        for(int i=;i<=cnt;i++)
        {
            if(num[i]>maxx){
                maxx=num[i];
                ans.clear();
                ans.push_back(i);
            }else if(num[i]==maxx)
            {
                ans.push_back(i);
            }
        }
        cout<<maxx<<endl;
        for(int i=;i<ans.size();i++)
        {
            cout<<dio[ans[i]]<<endl;
        }
    }
}