2019徐州网络赛G

2019徐州网络赛G

题意

给定s字符串,定义一个回文串的价值是这个回文串中不同字母的个数,求s中所有回文串的价值之和.

思路

马拉车加序列自动机.

代码

#include
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;
}