【CF17E】Palisection(回文树)
【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;
}