题意
给定s字符串,定义一个回文串的价值是这个回文串中不同字母的个数,求s中所有回文串的价值之和.
思路
马拉车加序列自动机.
代码
#include<bits/stdc++.h>
using namespace std;
const int maxn=610010;
char Ma[maxn];
int Mp[maxn];
int Next[maxn][30];
long long ans=0;
void Manacher(char s[],int len){
int l=0;
Ma[l++]='$';Ma[l++]='#';
for(int i=1;i<=len;i++) Ma[l++]=s[i],Ma[l++]='#';
Ma[l]=0;
int mx=0,id=0;
for(int i=0;i<l;i++){
Mp[i]=mx>i?min(Mp[2*id-i],mx-i):1;
while(Ma[i+Mp[i]]==Ma[i-Mp[i]]) Mp[i]++;
if(i+Mp[i]>mx) mx=i+Mp[i],id=i;
}
int pos=1;
ans=0;
for(int i=2;i<l;i++){
int L=i/2-Mp[i]/2+1;
for(int j=0;j<26;j++){
if(Next[pos][j]>=L&&Next[pos][j]!=0) ans+=Next[pos][j]-L+1;
}
// printf("ans=%d\n",ans);
if(Ma[i]=='#') pos++;
}
}
char s[maxn*2];
int main(){
scanf("%s",s+1);
int len=strlen(s+1);
for(int i=1;i<=len;i++){
for(int j=0;j<26;j++) Next[i][j]=Next[i-1][j];
Next[i][s[i]-'a']=i;
}
Manacher(s,len);
printf("%lld\n",ans);
//system("pause");
return 0;
}