[caioj1462][exkmp]回文串

来源:互联网 发布:淘宝管理中心在哪 编辑:程序博客网 时间:2024/05/21 09:37

题意

给出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;}
原创粉丝点击