hdu 6096 String(AC自动机)

转换还是比较巧妙的。

因为要求前缀后缀都包含的个数，所以可以把字符串a转换成a#a这样一个字符串，比如abca就转换成abca#abca

然后对于一组前缀a后缀b转换成b#a,比如ab ca,就是ca#ab，

然后对前缀后缀的串建立AC自动机，让主串去匹配，如上述例子，ca#ab满足为abca#abca的一个子串，也就是abca满足这个前缀后缀，所以问题，就转换成了典型的ac自动机匹配问题。

加个#的原因是为了只让后缀#前缀这种串能在AC自动机匹配到。

然后求答案的时候，需要对连接到自己的fail的位置累加一下，含义想一下就明白了。

代码：

#include <bits/stdc++.h>using namespace std;const int maxn=1e6+5+1e5;char str[maxn];char *s[maxn];int d[maxn], pos[maxn], ans[maxn], st[maxn], len[maxn];char s1[maxn], s2[maxn];struct acho{    int nex[maxn+10][29], cnt[maxn+10], fail[maxn+10], match[maxn+10], nl[maxn+10];    int L, root;    int  newnode()    {        for(int i=0; i<27; i++)nex[L][i]=-1;        cnt[L++]=0;        fail[L-1]=0;        return L-1;    }    void init()    {        L=0;        root=newnode();    }    int insert(char *a)    {        int now=root, i;        for(i=0; a[i]; i++)        {            if(nex[now][a[i]-'a']==-1)            {                nex[now][a[i]-'a']=newnode();                nl[L-1]=i+1;            }            now=nex[now][a[i]-'a'];        }        cnt[now]=1;        return now;    }    queue<int>que;    void build()    {        while(que.empty()==0)que.pop();        int now=root;        for(int i=0; i<27; i++)        {            if(nex[now][i]==-1)nex[now][i]=root;            else            {                fail[nex[now][i]]=root;                que.push(nex[now][i]);            }        }        while(que.empty()==0)        {            now=que.front();            que.pop();            for(int i=0; i<27; i++)            {                if(nex[now][i]!=-1)                {                    que.push(nex[now][i]);                    fail[nex[now][i]]=nex[fail[now]][i];                    match[nex[now][i]]=cnt[nex[now][i]]?nex[now][i]:match[nex[fail[now]][i]];                }                else                {                    nex[now][i]=nex[fail[now]][i];                }            }        }        return;    }    void search(char *a, int len)    {        int i, now=root;//        printf("%s\n", a);        for(i=0; a[i]; i++)        {            now=nex[now][a[i]-'a'];            while(nl[now]>len)            {                now=fail[now];            }            ans[match[now]]++;//            printf("%c %d\n", a[i], now);        }    }    void cal()    {        memset(d, 0, sizeof(int)*(L+1));        int j, k=0, i;        for(i=0; i<L; i++)d[fail[i]]++;        for(i=0; i<L; i++)if(!d[i])st[k++]=i;        for(i=0; i<k; i++)        {            j=fail[st[i]];            ans[j]+=ans[st[i]];            if(!(--d[j]))st[k++]=j;        }        return;    }}acho;int main(){    int t, i, j, n, q;    cin>>t;    while(t--)    {        acho.init();        scanf("%d%d", &n, &q);        j=0;        for(i=0; i<n; i++)        {            s[i]=str+j;            scanf("%s", s[i]);            len[i]=strlen(s[i])+1;            j+=len[i];            strcpy(str+j, s[i]);            str[j-1]='z'+1;            j+=len[i];        }        for(i=0; i<q; i++)        {            scanf("%s%s", s1+1, s2);            s1[0]='z'+1;            strcat(s2, s1);            pos[i]=acho.insert(s2);//            printf("%d ", pos[i]);        }//        printf("\n");        acho.build();//        for(i=0; i<acho.L; i++)printf("%d %d\n", i, acho.match[i]);        for(i=0; i<n; i++)        {            acho.search(s[i], len[i]);        }        acho.cal();        for(i=0; i<q; i++)printf("%d\n", ans[pos[i]]);        for(i=0; i<=acho.L; i++)        {            ans[i]=0;            acho.match[i]=0;        }    }    return 0;}