hdu 5853 Jong Hyok and String(广义后缀自动机)

题目链接：hdu 5853 Jong Hyok and String

题意：

给你n个字符串，m个询问，每次询问一个字符串

定义set(s)={(i,j)} 表示 s在第i个字符串中出现，且末尾位置为j。

对于一个询问，求set(Qi)=set(t) ，t串的数量。

题解：

如果是n=1,那么就是后缀自动机的一道裸题，答案就是Qi串匹配的最后一个节点x，ml[x]-ml[f[x]]。

现在是多个串，那么就建立一个广义后缀自动机。每次插入一个串后，将last=root，然后继续插下一个就行了。

最后匹配还是一样的操作。

#include<cstdio>
#include<cstring>
#define F(i,a,b) for(int i=a;i<=b;++i)
#define mst(a,b) memset(a,b,sizeof(a))
using namespace std;

const int N=1e5+7,tyn=26,M=N*2;
struct SAM{
    int tr[M][tyn],f[M],ml[M],ed,last,p,x,r,q;
    int cnt[M],b[M],d[M];
    inline int gid(char x){return x-'a';}
    inline void nc(int s,int &p){
        ml[p=++ed]=s,f[ed]=cnt[ed]=0,mst(tr[ed],0);
    }
    void clear(){ed=0,nc(0,last);}
    void add(int w){
        nc(ml[x=last]+1,p),last=p,cnt[p]=1;
        while(x&&!tr[x][w])tr[x][w]=p,x=f[x];
        if(!x)f[p]=1;
        else if(ml[x]+1==ml[q=tr[x][w]])f[p]=q;
        else{
            nc(ml[x]+1,r),f[r]=f[q],f[p]=f[q]=r;
            memcpy(tr[r],tr[q],sizeof(tr[r]));
            while(x&&tr[x][w]==q)tr[x][w]=r,x=f[x];
        }
    }
    void build(char *s){for(int i=1;s[i];i++)add(gid(s[i]));}
    
    int go(char *s,int x=1)//与s做匹配
    {
        int len=strlen(s+1);
        for(int i=1;i<=len;i++)
        {
            int w=gid(s[i]);
            if(tr[x][w])x=tr[x][w];
            else return 0;
        }
        return ml[x]-ml[f[x]];
    }
}sam;

char s[N];
int t,n,m,cas;

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        sam.clear();
        F(i,1,n)
        {
            sam.last=1;
            scanf("%s",s+1);
            sam.build(s);
        }
        printf("Case #%d:
",++cas);
        F(i,1,m)
        {
            scanf("%s",s+1);
            printf("%d
",sam.go(s));
        }
    }
    return 0;
}