题意:
给定字符串,一个(k-回文串)s定义为:
s串是回文串,且s[1,n/2]是((k-1)-回文串)
(非回文串是(0-回文串))
要求计算出给定字符串所有前缀的k,并输出他们的k的和
数据范围:n<=5e6
解法:
令d[i]表示s[1,i]的k值
显然d[i]=d[n/2]+1,需要满足s[1,n/2]与s[i-n/2+1,i]互为反串
正向负向都预处理一个hash就能快速判断是否是反串了
code:
#include<bits/stdc++.h>
using namespace std;
#define ull unsigned long long
const int maxm=5e6+5;
const int p=1331;
char s[maxm];
ull base[maxm];
ull h[maxm];
ull h2[maxm];
int d[maxm];
ull get(int l,int r){
return h[r]-h[l-1]*base[r-l+1];
}
ull get2(int l,int r){
return h2[l]-h2[r+1]*base[r-l+1];
}
signed main(){
scanf("%s",s+1);
int n=strlen(s+1);
base[0]=1;
for(int i=1;i<=n;i++){
base[i]=base[i-1]*p;
h[i]=h[i-1]*p+s[i];
}
for(int i=n;i>=1;i--){//反着
h2[i]=h2[i+1]*p+s[i];
}
//
d[1]=1;
for(int i=2;i<=n;i++){
int t=i/2;
ull a1=get(1,t),a2=get2(i-t+1,i);
if(a1==a2){
d[i]=d[t]+1;
}
}
int ans=0;
for(int i=1;i<=n;i++){
ans+=d[i];
}
cout<<ans<<endl;
return 0;
}