【CF17E】Palisection（回文樹）

題面

洛谷

題解

題意：
求有重疊部分的回文子串對的數量

所謂正難則反
求出所有不重疊的即可
求出以一個位置結束的回文串的數量
和以一個位置為開始的回文串的數量
然後對應的乘一下就行了
求法我用的是回文樹
維護每個節點到根節點的距離，
就是回文後綴的數量

CF上的空間是\(128MB\)
卡的很
所以所有的連邊考慮用鄰接表來代替

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
 

#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 2000020
#define MOD 51123987
int n,p1[MAX],p2[MAX],ans,dep[MAX];
char s[MAX];
struct Line{int v,next,w;}e[MAX];
int cnt=1
 
;
int h[MAX];
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
struct PT
{
    struct Node
    {
        int ff,len;
    }t[MAX];
    int tot,last;
    void init()
    {
        for(int i=0;i<=tot;++i)
        {
            h[i]=0;
            t[i].ff=t[i].len=0;
        }
        cnt=1
 
;
        last=0;
        t[tot=1].len=-1;
        t[0].ff=t[1].ff=1;
    }
    int nt(int k,int c)
    {
        for(int i=h[k];i;i=e[i].next)
            if(e[i].w==c)return e[i].v;
        return 0;
    }       
    void extend(int c,int n,char *s)
    {
        int p=last;
        while(s[n-t[p].len-1]!=s[n])p=t[p].ff;
        if(!nt(p,c))
        {
            int v=++tot,k=t[p].ff;
            while(s[n-t[k].len-1]!=s[n])k=t[k].ff;
            t[v].len=t[p].len+2;
            t[v].ff=nt(k,c);
            dep[v]=dep[t[v].ff]+1;
            Add(p,v,c);
        }
        last=nt(p,c);
    }
}pt1;
int main()
{
    scanf("%d",&n);
    scanf("%s",s+1);
    pt1.init();
    for(int i=1;i<=n;++i)
    {
        pt1.extend(s[i]-97,i,s);
        ans=(ans+(p1[i]=dep[pt1.last]))%MOD;
    }
    ans=1ll*ans*(ans-1)/2%MOD;
    reverse(&s[1],&s[n+1]);
    memset(dep,0,sizeof(dep));
    pt1.init();
    for(int i=1;i<=n;++i)
    {
        pt1.extend(s[i]-97,i,s);
        p2[n-i+1]=dep[pt1.last];
    }
    for(int i=n;i;--i)(p2[i]+=p2[i+1])%=MOD;
    for(int i=1;i<=n;++i)ans=(ans-1ll*p1[i]*p2[i+1]%MOD+MOD)%MOD;
    printf("%d\n",ans);
    return 0;
}

【CF17E】Palisection（回文樹）