Best Reward-(扩展kmp)

题目链接：点击进入

题目

题意

一个字符串，由26个小写英文字母组成, 每个字母都有各自的价值, 要你将一个字符串切成两个子串, 总价值为两个串价值和, 若是回文, 则串的价值为每个字母价值和, 否则为0，问最大价值多少。

思路

将原串逆序，然后原串与逆序串相互ekmp匹配；对逆串枚举分割点，同时判断逆串分隔的两部分是否回文（根据extend数组判断），价值相加，循环取最大值

代码

#include<iostream>
#include<string>
#include<map>
#include<set>
//#include<unordered_map>
#include<queue>
#include<cstdio>
#include<vector>
#include<cstring>
#include<algorithm>
#include<iomanip>
#include<cmath>
#include<fstream>
#define X first
#define Y second
#define best 131 
#define INF 0x3f3f3f3f3f3f3f3f
#define inf 0x3f3f3f3f
#define pii pair<int,int>
#define lowbit(x) x & -x
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const double eps=1e-7;
const double pai=acos(-1.0);
const int N=2e4+10;
const int maxn=1e6+10;
const int mod=1e9+7;
int t,n,m,k,cnt,tot;
int Next1[maxn],Next2[maxn],ext1[maxn],ext2[maxn],v[maxn],sum[maxn];
//扩展KMP是用来求主串每个点可以向后延伸与模式串匹配的最长的长度
void getnext(string &s,int Next[])//模式串 
{
    int f=0,l=0,l1=s.size();
    Next[0]=l1;
    for(int i=1;i<l1;i++) 
	{
        Next[i]=min(l-i+1,Next[i-f]);
        if(Next[i]<0) Next[i]=0;
        while(i+Next[i]<l1&&s[Next[i]]==s[i+Next[i]]) 
            Next[i]++;
        if(i+Next[i]-1>l) 
		{
            l=i+Next[i]-1;
            f=i;
        }
    }
}
void exkmp(string &s,string &t,int Next[],int ext[])// t是主串，s是模式串 
{
	int l1=s.size(),l2=t.size();
    getnext(s,Next);
    int f=0,l=-1,minn=min(l1,l2);
    while(l<minn-1&&t[l+1]==s[l+1])
        l++;
    ext[0]=l+1;
    for(int i=1;i<l2;i++)
	{
        ext[i]=min(l-i+1,Next[i-f]);
        if(ext[i]<0) ext[i]=0;
        while(ext[i]<l1&&i+ext[i]<l2&&s[ext[i]]==t[i+ext[i]])
            ext[i]++;
        if(i+ext[i]-1>l)
		{
            l=i+ext[i]-1;
            f=i;
        }
    }
}
int main( )
{
//	ios::sync_with_stdio(false);
//	cin.tie(0);cout.tie(0);
	cin>>t; 
    while(t--)
    {
    	string s1,s2;
        for(int i=1;i<=26;i++)
            cin>>v[i];
        cin>>s1;
        int len=s1.size();
        sum[0]=0;
        for(int i=0;i<len;i++)
        {
            s2+=s1[len-i-1];
            sum[i+1]=sum[i]+v[s2[i]-'a'+1];
        }
        exkmp(s1,s2,Next1,ext1);  //s1是模式串 s2是主串
        exkmp(s2,s1,Next2,ext2);  //s2是模式串 s1是主串
        int ans=0;
        for(int i=1;i<len;i++) 
        {
            int tmp=0,pos=len-i;
            if(i+ext2[i]>=len)//对串2来说,前len-i回文
                tmp+=sum[len-i];
            if(pos+ext1[pos]>=len)//对串2来说,后i个回文
                tmp+=sum[len]-sum[pos];
            if(tmp>ans)
                ans=tmp;
        }
        cout<<ans<<endl;
    }
	return 0;
}