hdu 5157 Harry and magic string(manacher+dp）

我们可以先用mancher算法对字符串进行处理，把以每个点为中心的回文串半径求出来，然后进行处理。

加入对以p为中心的点，从p-r[i]+1~p都是回文串的开头,那么对于每个回文串（开头是j）只要记录结尾从1~j-1的回文串个数，我们可以用dp记录以每个点为结尾的回文串个数，s[i]=sigma(dp[i]),则是结尾从1~j-1的回文串个数。那么对这个中心点来说一共的回文串对应该有：s[p-r[i]]+...+s[p-1]个，那么我们可以继续用一个数组s1[i]求s[i]的前缀和，那么总复杂度是O（n）。

至于dp[i]怎么求，你已经知道了半径，那从p~p+r[i]-1这些点的dp值都要加1，可以用树状数组来维护。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long LL;
char c[200005],f[100005];
int r[200005],bit[200005];
int d[200005];
LL s[200005],s1[200005];
int low(int n){return n&(-n);}
void merg(int p,int n,int k){
    while(p<=n){
        bit[p]+=k;
        p=p+low(p);
    }
}
int sum(int p){
    int s=0;
    while(p>0){
        s+=bit[p];
        p=p-low(p);
    }
    return s;
}
void mancher(int n){
    int i,id,mx;
    r[0]=1;
    mx=0;
    id=0;
    for(i=1;i<=2*n;i++){
        if(i>=mx) r[i]=1;
        else r[i]=min(r[id-(i-id)],mx-i);
        while(i-r[i]>=0&&i+r[i]<=2*n&&c[i-r[i]]==c[i+r[i]]) r[i]++;
        if(i+r[i]>mx){
            mx=i+r[i];
            id=i;
        }
    }
}
void get_back(int n){
    int i,j,p;
    memset(bit,0,sizeof(bit));
    memset(s,0,sizeof(s));
    memset(s1,0,sizeof(s1));
    for(i=1;i<=2*n;i++){
        p=i+r[i]-1;
        if(i%2==0){
            if(p>i){
                merg(i/2+1,n,1);
                merg(p/2+1,n,-1);
            }
        }
        else{
                merg((i+1)/2,n,1);
                merg(p/2+1,n,-1);
        }
    }
    s[0]=0;
    s1[0]=0;
    for(i=1;i<=n;i++){
        d[i]=sum(i);
        s[i]=s[i-1]+d[i];
        s1[i]=s1[i-1]+s[i];
    }
}
void work(int n){
    LL s=0;
    int i,j;
    for(i=1;i<=2*n;i++){
        if(i%2==0){
            if(r[i]>1){
                if((i-r[i]+1)/2!=0)
                s+=s1[i/2-1]-s1[(i-r[i]+1)/2-1];
                else s+=s1[i/2-1];
            }
        }
        else{
            if((i-r[i]+1)/2!=0)
            s+=s1[(i+1)/2-1]-s1[(i-r[i]+1)/2-1];
            else s+=s1[(i+1)/2-1];
        }
    }
    cout<<s<<endl;
}
int main()
{
    int i,j,n;
    while(scanf("%s",f)!=EOF){
        n=strlen(f);
        c[0]='#';
        for(i=1;i<=n;i++){
            c[2*i]='#';
            c[2*i-1]=f[i-1];
        }
        mancher(n);
        get_back(n);
        work(n);
    }
}