hdu 6096---String（AC自动机）

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <string>
#include <queue>
#include <vector>
using namespace std;
const int N=1e5+5;
string s[N];
struct Node{
    Node *son[30];
    Node *fail;
    int flag;
    int len;
}tr[6*N];
Node *root;
queue<Node*>Q;
int ans[N];
vector<int>v[N];
int tot;

void init()
{
    tot=0;
    memset(ans,0,sizeof(ans));
    root=&tr[0];
    while(!Q.empty()) Q.pop();
    for(int i=0;i<N;i++) v[i].clear();
    for(int i=0;i<6*N;i++)
    {
        tr[i].flag=0;
        tr[i].fail=NULL;
        for(int j=0;j<30;j++) tr[i].son[j]=NULL;
    }
}
void build(string s,int id)
{
    Node *now=root;
    for(int i=0;i<s.length();i++)
    {
       int x=s[i]-‘_‘;
       if(!now->son[x]) now->son[x]=&tr[++tot];
       now=now->son[x];
    }
    if(now->flag) {
        v[now->flag].push_back(id);
        return ;
    }
    now->flag=id;
    now->len=s.length();
}
void setFail()
{
    Q.push(root);
    while(!Q.empty())
    {
        Node *now=Q.front(); Q.pop();
        for(int i=0;i<30;i++)
        {
            if(now->son[i])
            {
                Node *p=now->fail;
                while(p && (!(p->son[i]))) p=p->fail;
                now->son[i]->fail=(p)?p->son[i]:root;
                Q.push(now->son[i]);
            }
            else now->son[i]=(now!=root)?now->fail->son[i]:root;
        }
    }
}
void query(string s)
{
    Node *now=root;
    int len=s.length();
    for(int i=0;i<len;i++)
    {
        int x=s[i]-‘_‘;
        now=now->son[x];
        Node *p=now;
        while(p!=root)
        {
            if(p->flag && p->len<=len/2+1) ans[p->flag]++;
            p=p->fail;
        }
    }
}
int main()
{
    int T; cin>>T;
    while(T--)
    {
        init();
        int n,q; scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++)
        {
            cin>>s[i];
            s[i]=s[i]+"_"+s[i];
        }
        for(int i=1;i<=q;i++)
        {
            string s1,s2; cin>>s1>>s2;
            string ss=s2+"_"+s1;
            build(ss,i);
        }
        setFail();
        for(int i=1;i<=n;i++)
        {
            query(s[i]);
        }
        for(int i=1;i<=q;i++)  ///处理相同的前后缀;
        {
            for(int j=0;j<v[i].size();j++)
                ans[v[i][j]]=ans[i];
        }
        for(int i=1;i<=q;i++) printf("%d\n",ans[i]);
    }
    return 0;
}