HDU5785 manacher+差分数组

用manacher算法O(n)求出所有的回文半径。有了回文半径后,就可以求出L[i]表示以i结尾的回文串的起始位置的和R[i]表示以i起始的回文串的结尾位置的和,然后就可以求出答案了,这里要注意奇偶长度回文串的不同处理。复杂度O(n)

 #include<bits/stdc++.h>
using namespace std;
const int N = 1e6+;
typedef long long ll;
const int mod = 1e9+;
int n,m,i,r,p,f[N<<]; char a[N],s[N<<];
ll L[N], R[N];
void add(ll& a, ll b){
a += b;
if(a >= mod||a <= -mod) a %= mod;
}
int main(){
while(~scanf("%s", a+)){
n = strlen(a+);
for(i = ; i <= n; i++) s[i<<] = a[i], s[i<<|] = '#';
s[] = '$', s[] = '#', s[m = (n+)<<] = '@';
for(r=p=,f[]=,i=;i<m;i++){
for(f[i]=r>i?min(r-i,f[p*-i]):;s[i-f[i]]==s[i+f[i]];f[i]++);
if(i+f[i]>r)r=i+f[i],p=i;
} memset(L, , sizeof(ll)*(n+));
memset(R, , sizeof(ll)*(n+));
for(i=;i<=*n; i++){
int ret = f[i]-, pos = i/;
if(ret == ) continue ;
ret /= ;
if(i&){
//[pos+1, pos+rer/2]
add(L[pos+], pos), add(L[pos+], --pos), add(L[pos+ret+], ret-pos), add(L[pos+ret+], pos+-ret);
//[pos-ret/2+1, pos]
add(R[pos-ret+], pos+ret), add(R[pos-ret+], --pos-ret), add(R[pos+], -pos), add(R[pos+], pos+);
}else{
//[pos, pos+ret/2]
add(L[pos], pos), add(L[pos+], --pos), add(L[pos+ret+], ret-pos+), add(L[pos+ret+], pos-ret);
//[pos-ret/2, pos]
add(R[pos-ret], pos+ret), add(R[pos-ret+], --pos-ret), add(R[pos+], -pos), add(R[pos+], pos);
}
}
for(i=; i<=n;i++)
add(L[i], L[i-]), add(R[i], R[i-]);
for(i=; i<=n;i++)
add(L[i], L[i-]), add(R[i], R[i-]);
ll ans = ;
for(i=;i<n;i++){
ans += L[i]*R[i+];
if(ans >= mod||ans <= -mod)
ans %= mod;
}
cout << (ans+mod)%mod << endl;
}
return ;
}
上一篇:codeforces 719E E. Sasha and Array(线段树)


下一篇:LeetCode算法题-Non-decreasing Array(Java实现)