2017多校第6场 HDU 6096 String AC自动机

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1000010;
const int maxs = 30;
string s[maxn];
int d[maxn],pos[maxn],ans[maxn],st[maxn],len[maxn];
struct Acautomata{
    int nxt[maxn][maxs],cnt[maxn],fail[maxn],match[maxn],dep[maxn];
    int L, root;
    int newnode(){
        for(int i=0; i<maxs; i++) nxt[L][i]=-1;
        cnt[L++]=0;
        fail[L-1]=0;
        return L-1;
    }
    void init(){
        L = 0;
        root = newnode();
    }
    int Insert(string a){
        int now=root;
        for(int i=0; a[i]; i++){
            if(nxt[now][a[i]-‘a‘]==-1){
                nxt[now][a[i]-‘a‘]=newnode();
                dep[L-1]=i+1;
            }
            now=nxt[now][a[i]-‘a‘];
        }
        cnt[now]=1;
        return now;
    }
    void build()
    {
        queue<int>q;
        int now=root;
        fail[0] = 0;
        for(int i=0; i<maxs; i++){
            if(nxt[now][i]==-1){
                nxt[now][i]=root;
                continue;
            }
            else{
                fail[nxt[now][i]]=root;
                q.push(nxt[now][i]);
            }
        }
        while(!q.empty())
        {
            now = q.front();
            q.pop();
            for(int i=0; i<maxs; i++){
                if(nxt[now][i]!=-1){
                    q.push(nxt[now][i]);
                    fail[nxt[now][i]] = nxt[fail[now]][i];
                    match[nxt[now][i]] = cnt[nxt[now][i]]?nxt[now][i]:match[nxt[fail[now]][i]];
                }else{
                    nxt[now][i] = nxt[fail[now]][i];
                }
            }
        }
    }
    void solve1(string a, int h){
        int now=root;
        int sz = a.size();
        for(int i=0; i<sz; i++){
            now=nxt[now][a[i]-‘a‘];
            while(dep[now]>h){
                now=fail[now];
            }
            ans[match[now]]++;
        }
    }
    void solve2(){//累加答案，对连到自己的fail进行累加求和
        memset(d, 0, sizeof(d));
        int j, k = 0;
        for(int i=0; i<L; i++) d[fail[i]]++;
        for(int i=0; i<L; i++) if(!d[i]) st[k++]=i;
        for(int i=0; i<k; i++){
            j=fail[st[i]];
            ans[j]+=ans[st[i]];
            if(!(--d[j])){
                st[k++]=j;
            }
        }
    }
}ZXY;
int n, q;
int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        ZXY.init();
        scanf("%d%d", &n,&q);
        for(int i=0; i<n; i++){
            cin>>s[i];
            len[i] = s[i].size()+1;
            string temp=s[i];
            s[i]+=char(‘z‘+1);
            s[i]+=temp;
        }
        for(int i=0; i<q; i++){
            string s1,s2;
            cin>>s1>>s2;
            s2 = s2 + char(‘z‘+1);
            s2 = s2 + s1;
            pos[i] = ZXY.Insert(s2);
        }
        ZXY.build();
        for(int i=0; i<n; i++){
            ZXY.solve1(s[i],len[i]);
        }
        ZXY.solve2();
        for(int i=0; i<q; i++) printf("%d\n", ans[pos[i]]);
        for(int i=0; i<=ZXY.L; i++){
            ans[i]=0;
            ZXY.match[i]=0;
        }
    }
    return 0;
}