HDU5785 manacher+差分数组

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

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
typedef long long ll;
const int mod = 1e9+7;
int n,m,i,r,p,f[N<<1]; char a[N],s[N<<1];
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+1)){
        n = strlen(a+1);
        for(i = 1; i <= n; i++) s[i<<1] = a[i], s[i<<1|1] = '#';
        s[0] = '$', s[1] = '#', s[m = (n+1)<<1] = '@';
        for(r=p=0,f[1]=1,i=2;i<m;i++){
            for(f[i]=r>i?min(r-i,f[p*2-i]):1;s[i-f[i]]==s[i+f[i]];f[i]++);
            if(i+f[i]>r)r=i+f[i],p=i;
        }

        memset(L, 0, sizeof(ll)*(n+5));
        memset(R, 0, sizeof(ll)*(n+5));
        for(i=2;i<=2*n; i++){
            int ret = f[i]-1, pos = i/2;
            if(ret == 0) continue ;
            ret /= 2;
            if(i&1){
                //[pos+1, pos+rer/2]
                add(L[pos+1], pos), add(L[pos+2], -1-pos), add(L[pos+ret+1], ret-pos), add(L[pos+ret+2], pos+1-ret);
                //[pos-ret/2+1, pos]
                add(R[pos-ret+1], pos+ret), add(R[pos-ret+2], -1-pos-ret), add(R[pos+1], -pos), add(R[pos+2], pos+1);
            }else{
                //[pos, pos+ret/2]
                add(L[pos], pos), add(L[pos+1], -1-pos), add(L[pos+ret+1], ret-pos+1), add(L[pos+ret+2], pos-ret);
                //[pos-ret/2, pos]
                add(R[pos-ret], pos+ret), add(R[pos-ret+1], -1-pos-ret), add(R[pos+1], 1-pos), add(R[pos+2], pos);
            }
        }
        for(i=1; i<=n;i++)
            add(L[i], L[i-1]), add(R[i], R[i-1]);
        for(i=1; i<=n;i++)
            add(L[i], L[i-1]), add(R[i], R[i-1]);
        ll ans = 0;
        for(i=1;i<n;i++){
            ans += L[i]*R[i+1];
            if(ans >= mod||ans <= -mod)
                ans %= mod;
        }
        cout << (ans+mod)%mod << endl;
    }
    return 0;
}