Description
由于一饭班长表示各种鸭梨,美丽的纪中决定历史性地启用第二饭堂。
而部分领导觉得,二饭依山傍水,环境优美,未免有不和谐的事情(你懂的)发生,决定到二饭巡视同学们用餐时的就座情况。
为了应付这一情况,同学们决定联合起来“布阵”。
方便起见,同学们已经把座位情况抽象成一个长度为n的仅含数字及字母的字符串,他们想请你帮忙算算这个字符串的和谐程度。
已知一个字符串被称为k-回文串的充要条件是它自身是回文串,并且它长为⌊n/2⌋(下取整)的前缀和后缀是(k-1)-回文串。根据定义,任意字符串(包括空串)都是0-回文串。
一个字符串的回文度数就是这个字符串的k的最大值。
而对于一个给定的字符串,它的和谐程度就是其所有前缀的回文度数之和。
你的任务就是算出这个和谐程度具体是多少。
Input
一行一个仅包含数字和字母的字符串。
Output
一行一个整数表示这个字符串的和谐高度。
Sample Input
abacaba
Sample Output
6
HINT
对于30%的数据字符串长度不超过1000
对于70%的数据字符串长度不超过100000
对于100%的数据字符串长度不超过5000000
Key To Problem
RKhash+dp吧,开始看到这道题卡内存,不敢开long long,之后有一个数就乘爆了,之后就挂了。
Code
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 5000010
#define MOD 1000000007
using namespace std;
typedef long long ll;
int n;
char s[N];
int pre[N],nxt[N],pow[N],pow10[N];
const ll QT=570000004;
ll f[N];
ll ans;
bool used[N];
void init()
{
pow[0]=1;
for(int i=1;i<=n;i++)
pow[i]=((ll)pow[i-1]*QT)%MOD;
}
int main()
{
// freopen("hw.in","r",stdin);
// freopen("hw.out","w",stdout);
scanf("%s",s+1);
n=strlen(s+1);
init();
int t=1;
for(int i=1;i<=n;i++)
{
pre[i]=((ll)pre[i-1]+(ll)s[i]*t)%MOD;
t=((ll)t*100)%MOD;
}
t=1;
for(int i=n;i>=1;i--)
{
nxt[i]=((ll)nxt[i+1]+(ll)s[i]*t)%MOD;
t=((ll)t*100)%MOD;
}
used[1]=1;
for(int i=2;i<=n;i++)
{
ll x=pre[i/2];
ll y=((ll)(nxt[(i+1)/2+1]-nxt[i+1]+MOD)%MOD)*(ll)pow[n-i]%MOD;
if(x==y)
used[i]=1;
}
for(int i=1;i<=n;i++)
if(used[i])
f[i]=f[i/2]+1,ans+=f[i];
cout<<ans<<endl;
return 0;
}