题意:给定 26 个字母的价值,再给定一个字符串 s,把 s 分成两段,对于这两段子串,若是回文,则累加字母价值和,若不是,则价值为零,求分段后价值最大为多少
分析:e-kmp,令 T=reverse(S) :
① S 当作母串,T当作模式串进行e-kmp。那么若 extend[k]+k=n,则表示 S[i~n-1] 是个回文串,为什么,extend[i] 表示S[i~n-1]和T串的最长公共前缀的长度,而 T 串又是 S 串的反转,令 extend[i]=len,则表示 S[i~n-1]=T[0~len-1],而 T[0~len-1] 又是 S[i~n-1] 的反转,由此得出 S[i~n-1] 是个回文串;
②同理,把 T 串当作母串,S当作模式串进行e-kmp,若 extend[k] + k=n,则表示 T[i~n-1] 是个回文串,即S[0~n-i]是个回文串。
综上,分别把S,T当作母串,另一个当作模式串,得到 extend1 数组和 extend2 数组,然后枚举中间的分割点就可以了,具体看代码体会。
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;
const int N = 5*1E5+10;
int n,v[30],sum[N];
int nxt1[N],nxt2[N];
int extend1[N],extend2[N];
char str1[N],str2[N];
void GetNxt(char s[],int len,int nxt[])
{
int i=0,j,po;
nxt[0]=len;
while(s[i]==s[i+1]&&i+1<len) i++; nxt[1]=i; //计算nxt[1]
po=1; //初始化po的位置
for(i=2;i<len;i++)
{
if(nxt[i-po]+i<nxt[po]+po) nxt[i]=nxt[i-po]; //第一种情况,可以直接得到nxt[i]的值
else //第二种情况,要继续匹配才能得到nxt[i]的值
{
j=nxt[po]+po-i;
if(j<0) j=0; //如果i>po+nxt[po],则要从头开始匹配
while(i+j<len&&s[j]==s[j+i]) j++; nxt[i]=j;
po=i; //更新po的位置
}
}
}
void ExKMP(char a[],int lena,char b[],int lenb,int nxt[],int extend[])
{
int i=0,j,po;
while(a[i]==b[i]&&i<lena&&i<lenb) i++; extend[0]=i;
po=0;
for(i=1;i<lena;i++)
{
if(nxt[i-po]+i<extend[po]+po) //第一种情况,直接可以得到Extendend[i]的值
extend[i]=nxt[i-po];
else //第二种情况,要继续匹配才能得到Extendend[i]的值
{
j=extend[po]+po-i;
if(j<0) j=0; //如果i>Extendend[po]+po则要从头开始匹配
while(i+j<lena&&j<lenb&&a[j+i]==b[j]) j++; extend[i]=j;
po=i; //更新po的位置
}
}
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
for(int i=0;i<26;i++) scanf("%d",&v[i]);
scanf("%s",str1);
n=strlen(str1);
for(int i=0;i<n;i++)
{
str2[i]=str1[n-i-1];
sum[i]=v[str1[i]-'a'];
}
for(int i=1;i<n;i++) sum[i]+=sum[i-1]; //计算 S 串的价值前缀和,优化
GetNxt(str2,n,nxt1);
ExKMP(str1,n,str2,n,nxt1,extend1);
GetNxt(str1,n,nxt2);
ExKMP(str2,n,str1,n,nxt2,extend2);
int ans=-1E8;
for(int i=1;i<n;i++)
{
int tot=0;
if(extend1[i]+i==n) tot+=sum[n-1]-sum[i-1];
if(extend2[n-i]+n-i==n) tot+=sum[i-1];
ans=max(ans,tot);
}
printf("%d\n",ans);
}
return 0;
}