hdu3613 扩展KMP

#include<stdio.h>
#include<string.h>
#define maxn 501000
char s[maxn],t[maxn];
int next[maxn],extand1[maxn],extand2[maxn];
int ans[30],sum[maxn];
void getnext(int *next,char *t)
{
    int i,k,j,len=strlen(t);
    next[0]=len;
    i=0;
    while(i<len-1&&t[i]==t[i+1])
    {
        i++;
    }
    next[1]=i;
    int a=1;
    for(k=2;k<len;k++)
    {
        int p=next[a]+a-1;
        int l=next[k-a];
        if(k-1+l>=p)
        {
            int j=p-k+1>0?p-k+1:0;
            while(k+j<len&&t[k+j]==t[j])
                j++;
            next[k]=j;
            a=k;
        }
        else next[k]=l;
    }
}
void ekmp(char *s,char *t,int *next,int *extand)
{
    int i,j,slen=strlen(s),tlen=strlen(t),k;
    getnext(next,t);//求next数组
    /*
    for(i=0;i<tlen;i++)
        printf("%d ",next[i]);
    printf("
");
    */
    int minlen=slen<tlen?slen:tlen;
    int a=0;
    while(a<minlen&&s[a]==t[a])    a++;
    extand[0]=a;
    a=0;
    
    for(k=1;k<slen;k++)
    {
        int p=a+extand[a]-1;
        int l=next[k-a];
        if(k-1+l>=p)
        {
            int j=p-k+1>0?p-k+1:0;
            while(k+j<slen&&j<tlen&&s[k+j]==t[j])
                j++;
            extand[k]=j;
            a=k;
        }
        else extand[k]=l;
    }
    /*for(i=0;i<slen;i++)
        printf("%d ",extand[i]);
    printf("
");*/
}
int main()
{
    int i,j,te;
    scanf("%d",&te);
    while(te--)
    {
        for(i=0;i<26;i++)
            scanf("%d",&ans[i]);
        scanf("%s",s);
        int len=strlen(s);
        for(i=0;i<len;i++)
        {
            t[len-i-1]=s[i];
            if(i==0)
                sum[i]=ans[s[i]-'a'];
            else
                sum[i]=sum[i-1]+ans[s[i]-'a'];
        }
        t[len]='';
        ekmp(s,t,next,extand1);//s做主串
        ekmp(t,s,next,extand2);//t做主串

        /*for(i=0;i<len;i++)
        {
            printf("%d ",extand1[i]);
        }
        printf("
");
        for(i=0;i<len;i++)
        {
            printf("%d ",extand2[i]);
        }
        printf("
");*/

        int max=-99999999;
        int ret;
        for(i=1;i<len;i++)
        {
            ret=0;
            if(extand1[i]+i==len)
            {
                ret+=sum[len-1]-sum[i-1];
            }
            if(extand2[len-i]==i)//len-i表示t中的位置
            {
                ret+=sum[i-1];
            }
            if(ret>max)
                max=ret;
        }
        printf("%d
",max);
    }
}