HDU-5343 MZL's Circle Zhou (後綴自動機)

HDU-5343 MZL's Circle Zhou (後綴自動機)

題意：從\(A,B\)兩串中各選出一個子串\(x,y\)(能夠爲空)，求能構成的不一樣串的個數

爲了不重複統計，咱們須要規定優先級

對於任意一個答案中的串，它的合法切分中，咱們使得\(|x|\)最大

對於\(A,B\)分別構建後綴自動機

咱們統計\(y\)串開頭字符是\(c\)的合法方案數，就必須使得\(x\)不存在\(+c\)的轉移，不然就意味着\(|x|\)不是最大的

固然\(B\)要倒着創建自動機才能肯定這些子串的開頭字符

注意一下統計空串的一些細節

#include<bits/stdc++.h>
using namespace std;

#define reg register
typedef long long ll;
typedef unsigned long long ull;
#define rep(i,a,b) for(int i=a,i##end=b;i<=i##end;++i)
#define drep(i,a,b) for(int i=a,i##end=b;i>=i##end;--i)

#define pb push_back
template <class T> inline void cmin(T &a,T b){ ((a>b)&&(a=b)); }
template <class T> inline void cmax(T &a,T b){ ((a<b)&&(a=b)); }

char IO;
template<class T=int> T rd(){
    T s=0;
    int f=0;
    while(!isdigit(IO=getchar())) if(IO=='-') f=1;
    do s=(s<<1)+(s<<3)+(IO^'0');
    while(isdigit(IO=getchar()));
    return f?-s:s;
}

const int N=200000+10;

int n;
char A[N],B[N];
int trans[N][26],link[N],len[N],lst,stcnt,End[N],ma[N];

ull Sum[N],Ans;

void Init(){
    link[0]=-1,len[0]=0;
    rep(i,0,stcnt) ma[i]=0,memset(trans[i],0,104);
    stcnt=lst=0;
}

void Extend(int c) {
    int cur=++stcnt,p=lst;
    End[cur]=len[cur]=len[p]+1;
    while(~p && !trans[p][c]) trans[p][c]=cur,p=link[p];
    if(p==-1) link[cur]=0;
    else {
        int q=trans[p][c];
        if(len[q]==len[p]+1) link[cur]=q;
        else {
            int clone=++stcnt;
            End[clone]=End[q];
            memcpy(trans[clone],trans[q],104);
            len[clone]=len[p]+1,link[clone]=link[q];
            while(~p && trans[p][c]==q) trans[p][c]=clone,p=link[p];
            link[cur]=link[q]=clone;
        }
    }
    lst=cur;
}


int main(){
    rep(kase,1,rd()) {
        scanf("%s%s",A+1,B+1);
        Init();
        rep(i,1,strlen(A+1)) Extend(A[i]-'a');
        memset(Sum,0,sizeof Sum),Ans=0;
        rep(i,1,stcnt) { 
            Ans+=len[i]-len[link[i]];// 考慮y爲空串
            rep(j,0,25) if(!trans[i][j]) Sum[j]+=len[i]-len[link[i]]; // 若是不存在轉移就累和
        }
        rep(i,0,25) if(!trans[0][i]) Sum[i]++; // 考慮x爲空串
        n=strlen(B+1);
        Init();
        drep(i,n,1) Extend(B[i]-'a'); // 要倒着創建自動機才能肯定這些子串的開頭字符
        rep(i,1,stcnt) if(End[i]) Ans+=(len[i]-len[link[i]])*Sum[B[n-End[i]+1]-'a'];
        Ans++;
        printf("%llu\n",Ans);
    }
}