hdu_2222_Keywords Search(AC自動機板子)

題目連接：hdu_2222_Keywords Search

存個自己寫的AC自動機

#include<cstdio>
#include<cstring>
#define F(i,a,b) for(int i=a;i<=b;i++)

const int AC_N=10001*50,tyn=26;//數量乘串長,類型數量
struct AC_automation{
    int tr[AC_N][tyn],cnt[AC_N],Q[AC_N],fail[AC_N],tot;
	inline int getid(char x){return x-'a';}
    void nw(){cnt[++tot]=0;memset(tr[tot],-1,sizeof(tr[tot]));}
    void init(){tot=-1,fail[0]=-1,nw();}
    void insert(char *s,int x=0){
        for(int len=strlen(s),i=0,w;i<len;x=tr[x][w],i++)
            if(tr[x][w=getid(s[i])]==-1)nw(),tr[x][w]=tot;
        cnt[x]++;//串尾標記
    }
    void build(int head=1,int tail=0){
        for(Q[++tail]=0;head<=tail;){
            for(int i=0,x=Q[head++],p=-1;i<tyn;i++)if(~tr[x][i]){
                if(x==0)fail[tr[0][i]]=0;
                else for(p=fail[x],fail[tr[x][i]]=0;~p;p=fail[p])
                        if(~tr[p][i]){fail[tr[x][i]]=tr[p][i];break;}
                Q[++tail]=tr[x][i];
            }
        }
    }
    int ask(char *s,int ans=0){
        for(int len=strlen(s),w,i=0,x=0,j;i<len;i++){
            while(tr[x][w=s[i]-'a']==-1&&x)x=fail[x];
            x=tr[x][w],x=(~x)?x:0,j=x;//加cnt後標記爲0，防止重複計數
            while(j&&cnt[j])ans+=cnt[j],cnt[j]=0,j=fail[j];
        }
        return ans;
    }
}AC;
char buf[1000010];
int main(){
    int t,n;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        AC.init();
        F(i,1,n)scanf("%s",buf),AC.insert(buf);
        AC.build();
        scanf("%s",buf);
        printf("%d\n",AC.ask(buf));
    }
    return 0;
}