HDU 3613 Manacher算法

题意

有一串字符，将其切成两串，如果一串是回文的，那么便是这个回文串所有的元素值累加。如果不是回文的，那么值便是0。问最大可获得的值是多少。

题解

Manacher模板套上去，记录一下前缀回文和后缀回文。枚举切割点，求最大值。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define LL long long
#define INF 0x3f3f3f3f3f3f3f3f
#define MAXN 120050
#define MOD 1000000007
#define EPS 1e-3
using namespace std;
char s[520010],ts[1020010];
int p[1020010],mp[1020010];
int val[150];
bool pre[520010],last[520010];
int sum[520010];

void solve() {
    MEM(mp,-1);
    MEM(p,0);
    MEM(pre,false);
    MEM(last,false);
    int n=strlen(s);
    int pos=0;
    sum[0]=val[s[0]];
    UP(i,1,n) sum[i]=sum[i-1]+val[s[i]];
//    cout<<sum[n-1]<<endl;
    ts[pos++]='$';
    mp[pos]=0;
    ts[pos++]='#';
    UP(i,0,n) {
        mp[pos]=i;
        ts[pos++]=s[i];
        mp[pos]=i;
        ts[pos++]='#';
    }
    ts[pos]='\0';
    int mx=0,id;
    UP(i,0,pos) {
        if(mx>i) p[i]=min(p[id*2-i],mx-i);
        else p[i]=1;
        W(ts[i+p[i]]==ts[i-p[i]]) p[i]++;
        if(i+p[i]>mx) {
            mx=i+p[i];
            id=i;
        }
        if(p[i]==i) if(mp[i+p[i]-1]!=-1) pre[mp[i+p[i]-1]]=true;
        if(i+p[i]==pos) if(mp[i-p[i]+2]!=-1) last[mp[i-p[i]+2]]=true;
    }
    int ans=-INF;
    UP(i,0,n-1) {
        int tmp=0;
        if(pre[i]) tmp+=sum[i];
        if(last[i+1]) tmp+=(sum[n-1]-sum[i]);
//        cout<<i<<" "<<tmp<<" "<<pre[i]<<" "<<last[i+1]<<endl;
        ans=max(ans,tmp);
    }
    printf("%d\n",ans);
}

int main() {
    int t;
    scanf("%d",&t);
    W(t--) {
        UP(i,'a','z'+1) scanf("%d",&val[i]);
        scanf("%s",s);
        solve();
    }
}