HDU-3613 Best Reward

题目链接：https://vjudge.net/problem/HDU-3613

参考别人的做法：https://www.2cto.com/kf/201211/165735.html

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
const int N=5*1e5+10; //文本串长度
const int M=5*1e5+10; //模式串长度
char a[N]; //文本串
char b[M]; //模式串
int next0[M],extend[N],extend0[N];
int value[26];
void get_next() 
{
    int len=strlen(b);
    next0[0]=len;
    int i=0;
    while(i<len-1&&b[i]==b[i+1]) ++i;
    next0[1]=i;
    i=1;
    for(int k=2;k<len;++k)
	{
        int p=i+next0[i]-1;
        int L=next0[k-i];
        if(k-1+L>=p) 
		{
            int j=(p-k+1)>0?(p-k+1):0;
            while(k+j<len&&b[k+j]==b[j]) ++j;
            next0[k]=j;
            i=k;
        } 
		else next0[k]=L;
    }
}
void get_extend() //extend[i]保存以a[i]为起始的文本串与模式串的最大公共前缀长度 
{
    int i=0,j,po,alen=strlen(a),blen=strlen(b);
    get_next();
    while(a[i]==b[i]&&i<alen&&i<blen) i++;
    extend[0]=i;
    po=0;
    for(i=1;i<alen;i++) 
	{
        if(next0[i-po]+i<extend[po]+po) extend[i]=next0[i-po];
		else 
		{
            j=extend[po]+po-i;
            if(j<0) j=0;
            while(i+j<alen&&j<blen&&a[j+i]==b[j]) j++;
            extend[i]=j;
            po=i;
        }
    }
}
int main()
{
	int T;
	scanf("%d",&T);
	while(T--)
	{
		memset(b,'\0',sizeof(b));
		for(int i=0;i<26;i++) scanf("%d",&value[i]);
		scanf("%s",a);
		int len=strlen(a);
		for(int i=0;i<len;i++)
		b[i]=a[len-i-1];
		get_extend();
		memcpy(extend0,extend,sizeof(extend));
		strcpy(a,b);
		for(int i=0;i<len;i++)
		b[i]=a[len-i-1];
		get_extend();
		int ans=-1000000000;
		int sum=0,t=0;
		for(int i=0;i<len;i++)
		sum+=value[a[i]-'a'];
		t+=value[b[0]-'a'];
		for(int i=1;i<len;i++)
		{
			int k=0;
			if(extend0[i]+i==len) k+=sum-t;
			if(extend[len-i]==i) k+=t;
			ans=max(ans,k);
			t+=value[b[i]-'a'];
		}
		printf("%d\n",ans);
	}
	return 0;
}