hdu 5343 MZL's Circle Zhou(后缀自动机)

题目链接：hdu 5343 MZL's Circle Zhou

题意：

给你两个串A，B，问从A,B中选子串x,y，问x+y可以组成多少个不同的串，x和y可以为空。

题解：

贴一个官方的题解

 

#include<bits/stdc++.h>
#define mst(a,b) memset(a,b,sizeof(a))
#define F(i,a,b) for(int i=(a);i<=(b);++i)
using namespace std;
using ll=unsigned long long;

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 b[M],d[M]; ll cnt[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 upright(int mx=0){
        F(i,0,ed)d[i]=0;
        F(i,1,ed)d[mx<ml[i]?mx=ml[i]:ml[i]]++;
        F(i,1,mx)d[i]+=d[i-1];
        F(i,1,ed)b[d[ml[i]]--]=i;
    }
    void build(char *s){for(int i=1;s[i];i++)add(gid(s[i]));}
}sam[2];


char s[N];
int t;

void work()
{
    for(int i=sam[1].ed;i;--i)
    {
        int x=sam[1].b[i];
        sam[1].cnt[x]=1;
        F(j,0,25)
        {
            int v=sam[1].tr[x][j];
            if(v)sam[1].cnt[x]+=sam[1].cnt[v];
        }
    }
}

void solve()
{
    for(int i=sam[0].ed;i;--i)
    {
        int x=sam[0].b[i];
        sam[0].cnt[x]=0;
        F(j,0,25)
        {
            int v=sam[0].tr[x][j];
            if(v)sam[0].cnt[x]+=sam[0].cnt[v];
            else sam[0].cnt[x]+=sam[1].cnt[sam[1].tr[1][j]];
        }
    }
    ll ans=sam[0].cnt[1];//x为空的时候已经统计过了
    F(i,2,sam[0].ed)ans+=sam[0].ml[i]-sam[0].ml[sam[0].f[i]];//y为空的时候
    printf("%llu\n",ans+1);//+1表示x和y都为空的时候
}

int main(){
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s+1);
        sam[0].clear(),sam[0].build(s),sam[0].upright();
        scanf("%s",s+1);
        sam[1].clear(),sam[1].build(s),sam[1].upright();
        work();solve();
    }
    return 0;
}