HDU2222 Keywords Search [AC自动机]

　　AC自动机的模版题，应该是很多人的AC自动机处女作吧。。

　　在斌牛的讲解下，总算是对AC自动机了解了个大概，这里采用了AC自动机和Trie图两种写法，据说后者在AC自动机DP中用的比较多，应该很快就会做到了。。。两者差不多，Trie图利用失败指针来建图，据说AC自动机到Trie图就是转化成了确定性有穷自动机。

　　贴一下两种写法的代码，当模版了。

　AC自动机：

#include <string.h>
#include <stdio.h>
#include <queue>
#define MAXS 500010

int cas,n;
char s[55],find[1000005];
int next[MAXS][26],pos,tot[MAXS],fail[MAXS];
int newnode(){
    memset(next[pos],0,sizeof next[pos]);
    tot[pos]=fail[pos]=0;
    return pos++;
}
void insert(char *s){
    int len=strlen(s);
    int p=0;
    for(int i=0;i<len;i++){
        int k=s[i]-'a';
        p=next[p][k]?next[p][k]:next[p][k]=newnode();
    }
    ++tot[p];
}
void makefail(){
    std::queue<int> q;
    q.push(0);
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=0;i<26;i++){
            int v=next[u][i];
            if(v==0)continue;
            q.push(v);
            if(u==0)continue;
            fail[v]=next[0][i];
            for(int f=fail[u];f!=0;f=fail[f]){
                if(next[f][i]){
                    fail[v]=next[f][i];
                    break;
                }
            }
        }
    }
}
int makeans(char *s){
    int ans=0;
    for(int i=0,p=0;s[i];i++){
        int k=s[i]-'a';
        while(p>0&&!next[p][k])p=fail[p];
        p=next[p][k];
        for(int t=p;t!=0;t=fail[t]){
            if(tot[t]>=0){
                ans+=tot[t];
                tot[t]=-1;
            }else break;
        }
    }
    return ans;
}
int main(){
freopen("test.in","r",stdin);
    scanf("%d",&cas);
    for(int ca=1;ca<=cas;ca++){
        scanf("%d",&n);
        pos=0;newnode();
        for(int i=1;i<=n;i++){
            scanf("%s",s);
            insert(s);
        }
        makefail();
        scanf("%s",find);
        int x=makeans(find);
        printf("%d\n",x);
    }
    return 0;
}

Trie图：

#include <string.h>
#include <stdio.h>
#include <queue>
#define MAXS 500010

int cas,n;
char s[55],find[1000005];
int next[MAXS][26],pos,tot[MAXS],fail[MAXS];
int newnode(){
    memset(next[pos],0,sizeof next[pos]);
    tot[pos]=fail[pos]=0;
    return pos++;
}
void insert(char *s){
    int p=0;
    for(int i=0;s[i];i++){
        int k=s[i]-'a';
        p=next[p][k]?next[p][k]:next[p][k]=newnode();
    }
    ++tot[p];
}
void makefail(){
    std::queue<int> q;
    q.push(0);
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=0;i<26;i++){
            int v=next[u][i];
            if(v==0)next[u][i]=next[fail[u]][i];
            else q.push(v);
            if(v!=0&&u!=0)fail[v]=next[fail[u]][i];
        }
    }
}
int makeans(char *s){
    int ans=0;
    for(int i=0,p=0,j;s[i];i++){
        int k=s[i]-'a';
        p=next[p][k];
        for(int j=p;j&&tot[j]!=-1;j=fail[j]){
            ans+=tot[j];
            tot[j]=-1;
        }
    }
    return ans;
}
int main(){
freopen("test.in","r",stdin);
    scanf("%d",&cas);
    for(int ca=1;ca<=cas;ca++){
        scanf("%d",&n);
        pos=0;newnode();
        for(int i=1;i<=n;i++){
            scanf("%s",s);
            insert(s);
        }
        makefail();
        scanf("%s",find);
        int x=makeans(find);
        printf("%d\n",x);
    }
    return 0;
}