[caioj1462][exkmp]回文串

题意

给出26个字母所代表的权值和一个字符串，要求把字符串分成两段（每一段长度至少为1，也就是必须要有字符），假如这一段子串是一个回文串，那么加上该串所有字符权值之和，求最大的权值和。

输入格式

输入一个整数T，表示数据组数
每组数据第一行输入26个数，表示26个字母的权值，第二行输入一个字符串（保证字符串内全是小写字母,2<=字符串长度<=500000）

输出格式

输出每组数据的最大权值和

样例输入

2
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
aba
1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1
acacac

样例输出

1
6

题解

好玄妙的exkmp。。其实回文串的话大佬们估计想的都是manacher，然鹅蒟蒻不会就exkmp了
由于extend[i]表示母串中第i位开始与子串中第1位开始的最长公共前缀
那么可以考虑一种做法，就是子串为母串反转，然后exkmp
这样一来，只要extend[i]*2大于要求的串的长度，那么一定是回文串
例如acbab
反转babca
可以观察到bab是个回文串，那么extend[3]是等于3的，倍2大于长度，那么他是个回文串
事实也是这样
做法就很简单啦~先反转，母串子串做一次exkmp，再用子串和母串做exkmp
因为你一个串分开来，前面要判是不是回文，后面也要判啊
好开心自己a了一道题

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
char s1[510000],s2[510000];
int p1[510000],p2[510000];
int extend1[510000],extend2[510000];
int sum[510000],a[27];
int lenx,leny;
void exkmp1()
{
    int x,k;
    p1[1]=leny;
    x=1;
    while(s2[x]==s2[x+1])x++;
    p1[2]=x-1;k=2;
    for(int i=3;i<=leny;i++)
    {
        int pp=k+p1[k]-1;
        int L=p1[i-k+1];
        if(i+L-1<pp)p1[i]=L;
        else
        {
            int j=pp-i+1;
            if(j<0)j=0;
            while(s2[j+1]==s2[j+i] && j+i<=leny)j++;
            p1[i]=j;
            k=i;
        }
    }
    x=1;
    while(s1[x]==s2[x] && x<=leny)x++;
    extend1[1]=x-1;k=1;
    for(int i=2;i<=lenx;i++)
    {
        int pp=k+extend1[k]-1;
        int L=p1[i-k+1];
        if(i+L-1<pp)extend1[i]=L;
        else
        {
            int j=pp-i+1;
            if(j<0)j=0;
            while(s2[j+1]==s1[j+i] && j+i<=lenx && j<=leny)j++;
            extend1[i]=j;
            k=i;
        }
    }
}
void exkmp2()
{
    int x,k;
    p2[1]=leny;
    x=1;
    while(s1[x]==s1[x+1])x++;
    p2[2]=x-1;k=2;
    for(int i=3;i<=lenx;i++)
    {
        int pp=k+p2[k]-1;
        int L=p2[i-k+1];
        if(i+L-1<pp)p2[i]=L;
        else
        {
            int j=pp-i+1;
            if(j<0)j=0;
            while(s1[j+1]==s1[j+i] && j+i<=leny)j++;
            p2[i]=j;
            k=i;
        }
    }
    x=1;
    while(s1[x]==s2[x] && x<=lenx)x++;
    extend2[1]=x-1;k=1;
    for(int i=2;i<=leny;i++)
    {
        int pp=k+extend2[k]-1;
        int L=p2[i-k+1];
        if(i+L-1<pp)extend2[i]=L;
        else
        {
            int j=pp-i+1;
            if(j<0)j=0;
            while(s1[j+1]==s2[j+i] && j+i<=leny && j<=lenx)j++;
            extend2[i]=j;
            k=i;
        }
    }
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(p1,0,sizeof(p1));memset(p2,0,sizeof(p2));
        memset(extend1,0,sizeof(extend1));memset(extend2,0,sizeof(extend2));
        for(int i=1;i<=26;i++)scanf("%d",&a[i]);
        scanf("%s",s1+1);lenx=leny=strlen(s1+1);
        sum[0]=0;
        for(int i=1;i<=lenx;i++)
        {
            sum[i]=sum[i-1]+a[s1[i]-'a'+1];
            s2[lenx-i+1]=s1[i];
        }
        exkmp1();exkmp2();
        int ans=0;
        for(int i=1;i<lenx;i++)
        {
            int m=0;
            if(extend2[lenx-i+1]*2>=i)m+=sum[i];
            if(extend1[i+1]*2>=(lenx-i+1))m+=sum[lenx]-sum[i];
            ans=max(ans,m);
        }
        printf("%d\n",ans);
    }
    return 0;
}