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
#include
#include
#include
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<


你可能感兴趣的:(动态规划,字符串)