hdu_5507_GT and strings(AC自动机)

题目链接：hdu_5507_GT and strings

题意：给n个字符串和q个询问，每个询问给两个数字x,y，问1.x是否为y的子序列，2.x是否为y的子串，是输出1，否则输出0，每个询问输出2个数字

题解：

对于子序列，朴素的做法，每次询问的复杂度为max(str[x],str[y])，题目好像有数据卡这个做法，反正会T，正解应该是建立一个序列自动机，首先将所有的字符串连续存起来，用数组来保存每个字符串的位置，然后dp[i][j]表示第i个字符的下一个字符的位置，询问的时候就能做到复杂度为min(str[x],str[y])。、

对于子串，可以用AC自动机来预处理，子串的询问有点特别，不是一般的在树上走，而是对每个节点进行寻找，然后用map来记录关系，不过在询问的时候也有个优化，不加这个优化一样会T，加了瞬间变成170+ms,具体看代码。

#include<bits/stdc++.h>
#define F(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef pair<int,int>P;

const int N=1e5+7;
int t,n,m,L[N],R[N],loc[N],dp[N][26],Q[N][2],ans[N][2];
char str[N];
map<P,int>ok;
const int AC_N=N,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,fail[tot]=0;memset(tr[tot],0,sizeof(tr[tot]));}
    void init(){tot=-1,fail[0]=-1,nw();}
    void insert(int l,int r,int x=0){
        for(int i=l,w;i<=r;x=tr[x][w],loc[i]=x,i++)
            if(!tr[x][w=getid(str[i])])nw(),tr[x][w]=tot;
        cnt[x]++;//串尾标记
    }
    void build(int head=1,int tail=0){
        for(int i=0;i<tyn;i++)if(tr[0][i])Q[++tail]=tr[0][i];
        while(head<=tail)for(int x=Q[head++],i=0;i<tyn;i++)
            if(tr[x][i])fail[tr[x][i]]=tr[fail[x]][i],Q[++tail]=tr[x][i];
            else tr[x][i]=tr[fail[x]][i];
    }
    void ask(int l,int r)
    {
        F(i,l,r)for(int p=loc[i];p;p=fail[p])
            if(cnt[p])ok[P(p,loc[r])]=1;
            else if(fail[p]&&!cnt[fail[p]])fail[p]=fail[fail[p]];//决定是否T的优化
    }
}AC;

void work1()
{
    F(i,0,25)dp[R[n]+1][i]=N;//建立“子序列自动机”
    for(int i=R[n];i>=1;i--)F(j,0,25)
        if(str[i]==j+'a')dp[i][j]=i;
            else dp[i][j]=dp[i+1][j];
    F(i,1,m)
    {
        int x=Q[i][0],y=Q[i][1];
        if(R[x]-L[x]>R[y]-L[y])ans[i][0]=0;
        else
        {    
            ans[i][0]=1;
            for(int j=L[x],p=L[y];j<=R[x];j++,p++)
            {
                p=dp[p][str[j]-'a'];
                if(p>R[y]){ans[i][0]=0;break;}
            }
        }
    }
}

void work2()
{
    AC.init(),ok.clear();
    F(i,1,n)AC.insert(L[i],R[i]);
    AC.build();
    F(i,1,n)AC.ask(L[i],R[i]);
    F(i,1,m)ans[i][1]=ok[P(loc[R[Q[i][0]]],loc[R[Q[i][1]]])];
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        int ed=1;
        F(i,1,n)
        {
            scanf("%s",str+ed);
            int len=strlen(str+ed);
            L[i]=ed,R[i]=ed+len-1,ed=R[i]+1;
        }
        F(i,1,m)scanf("%d%d",&Q[i][0],&Q[i][1]);
        work1(),work2();
        F(i,1,m)printf("%d%d",ans[i][0],ans[i][1]);
        puts("");
    }
    return 0;
}