题目链接:点击这里
因为长度为n的字符串的本质不同的回文串的级别是O(n)的。
所以我们直接跑回文自动机然后枚举所有本质不同的子回文串即可。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<cmath>
#define ll long long
#define llu unsigned ll
using namespace std;
const double eps = 1e-8;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const ll lnf=0x3f3f3f3f3f3f3f3f;
const int maxn=300015 ;
const int N=26 ;
char str[maxn];
int ha[100];
ll ans=0;
struct Palindromic_Tree {
int next[maxn][N];
int fail[maxn];
int cnt[maxn];
int num[maxn];
int len[maxn];
int S[maxn];
int last;
int n;
int p;
int newnode(int l)
{
for (int i=0;i<N;++i) next[p][i]=0 ;
cnt[p]=0 ;
num[p]=0 ;
len[p]=l ;
return p++ ;
}
void init()
{
p=0;
newnode(0);
newnode (-1);
last=0;
n=0;
S[n]=-1;
fail[0]=1;
}
int get_fail(int x)
{
while(S[n-len[x]-1]!=S[n]) x=fail[x];
return x;
}
void add(int c)
{
c-='a';
S[++ n]=c;
int cur=get_fail(last);
if (!next[cur][c])
{
int now=newnode(len[cur]+2);
fail[now]=next[get_fail(fail[cur])][c];
next[cur][c]=now;
num[now]=num[fail[now]]+1;
}
last=next[cur][c];
cnt[last]++;
}
void _count()
{
for(int i=p-1;i>=0;--i) cnt[fail[i]]+=cnt[i];
}
} pam;
void dfs(int x,int cnt)
{
for(int i=0;i<26;i++)
{
int pm=cnt;
if(pam.next[x][i])
{
ha[i]++;
if(ha[i]==1) pm=cnt+1;
ans+=(ll)pm*pam.cnt[pam.next[x][i]];
dfs(pam.next[x][i],pm);
ha[i]--;;
}
}
}
int main(void)
{
scanf("%s",str);
int len=strlen(str);
memset(ha,0,sizeof(ha));
ans=0;
pam.init();
for(int i=0;i<len;i++)
pam.add(str[i]);
pam._count();
dfs(0,0);
dfs(1,0);
printf("%lld\n",ans);
return 0;
}