hdu 3613回文関係

4926 ワード

http://acm.hdu.edu.cn/showproblem.php?pid=3613
26個の整数を与えて、アルファベットの26文字に対応する価値を与えて、それから一つの列を与えて、これを二つに切ってください.回文列ならその価値を計算して、最大の価値を聞きます.
manaacherでは、すべてのプレフィックスをマークし、最後に最大値を検索します.
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <cmath>
using namespace std;

const int MAXN=1e6+7;
const int oo=1e9+7;
char str[MAXN];
int p[MAXN],val[27],sum[MAXN];
int left[MAXN],right[MAXN];

/* str[]       ,          p[] p[i]    i          (        )、 sum[] sum[i]   i         left[] left[i]         i          right[] right[i]         i          */

void Manacher(char st[], int N)
{
    int id=0,po=0;
     for(int i=2;i<N;i++)
     {
         if(id>i)
         p[i]=min(id-i,p[2*po-i]);
         else
         p[i]=1;
         while(st[i-p[i]]==st[i+p[i]])
         p[i]++;
         if(p[i]+i>id)
         {
             id=p[i]+i;
             po=i;
         }
        if(p[i]==i)
            left[p[i]-1]=1;
        if(p[i]+i-1==N)
            right[p[i]-1]=1;
    }
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int i;
        memset(left,0,sizeof(left));
        memset(right,0,sizeof(right));
        for(i=0; i<26; i++)
        scanf("%d",&val[i]);
        scanf("%s",str);
        int len=strlen(str);
        for(i=1;i<=len;i++)
        sum[i]=sum[i-1]+val[str[i-1]-'a'];
        for(i=len;i>=0;i--)
        {
            str[i+i+2]=str[i];
            str[i+i+1]='#';
        }
        str[0]='$';
        Manacher(str,len+len+1);
        int ans=-oo;
        for(i=1;i<len;i++)
        {
            int tmp=0;
            if(left[i]==1)
            tmp+=sum[i];
            if(right[len-i]==1)
            tmp+=sum[len]-sum[i];
            ans=max(ans,tmp);
        }
        printf("%d
"
,ans); } return 0; }