HDU 3613 Manacher算法

来源:互联网 发布:三级分销佣金软件 编辑:程序博客网 时间:2024/05/18 03:35

题意

有一串字符,将其切成两串,如果一串是回文的,那么便是这个回文串所有的元素值累加。如果不是回文的,那么值便是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-3using 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();    }}