hdu3613 Best Reward(KMP)

本题难度在于判断回文前缀和回文后缀。我们设原串为s1，把原串翻转后得到的串记为s2.以s1为模式串去匹配s2，最后匹配到的位置我们记为k，k就是最长回文前缀长度。根据fail的性质，每一次fail[k]到的位置都满足前缀回文。那么我们就把这些回文前缀都标记了。回文后缀同理，我们以s2为模式串去匹配s1，那么k就是s2的最长回文前缀长度，即s1的最长回文后缀长度。同样处理，把这些回文后缀标记。然后我们就只需要线性扫描一遍在i后切一刀，更新最大值即可。

#include <cstdio>
#include <cstring>
#define N 500010
#define inf 0x3f3f3f
int tst,val[255],m,fail[N],sum[N];
bool prefix[N],suffix[N];
char s1[N],s2[N];
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x*f;
}
inline void getfail(char *t){
	int k=0;fail[1]=0;
	for(int i=2;i<=m;++i){
		while(k&&t[k+1]!=t[i]) k=fail[k];
		if(t[k+1]==t[i]) ++k;
		fail[i]=k;
	}
}
inline int kmp(char *s,char *t){
	int k=0;
	for(int i=1;i<=m;++i){
		while(k&&t[k+1]!=s[i]) k=fail[k];
		if(t[k+1]==s[i]) ++k;
	}return k;
}
int main(){
//	freopen("a.in","r",stdin);
	tst=read();
	while(tst--){
		memset(prefix,0,sizeof(prefix));
		memset(suffix,0,sizeof(suffix));
		for(int i='a';i<='z';++i) val[i]=read();
		scanf("%s",s1+1);m=strlen(s1+1);
		for(int i=1;i<=m;++i) sum[i]=sum[i-1]+val[s1[i]];
		for(int i=1;i<=m;++i) s2[m-i+1]=s1[i];
		getfail(s1);int k=kmp(s2,s1);
		for(;k;k=fail[k]) prefix[k]=1;
		getfail(s2);k=kmp(s1,s2);
		for(;k;k=fail[k]) suffix[m-k+1]=1;
		int ans=-inf;
		for(int i=1;i<m;++i){//cut between i,i+1;
			int res=0;
			if(prefix[i]) res+=sum[i];
			if(suffix[i+1]) res+=sum[m]-sum[i];
			if(res>ans) ans=res;
		}
		printf("%d\n",ans);
	}
	return 0;
}