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[],sum[maxn];
void getnext(int *next,char *t)
{
    int i,k,j,len=strlen(t);
    next[]=len;
    i=;
    while(i<len-&&t[i]==t[i+])
    {
        i++;
    }
    next[]=i;
    int a=;
    for(k=;k<len;k++)
    {
        int p=next[a]+a-;
        int l=next[k-a];
        if(k-+l>=p)
        {
            int j=p-k+>?p-k+:;
            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("\n");
    */
    int minlen=slen<tlen?slen:tlen;
    int a=;
    while(a<minlen&&s[a]==t[a])    a++;
    extand[]=a;
    a=;

    for(k=;k<slen;k++)
    {
        int p=a+extand[a]-;
        int l=next[k-a];
        if(k-+l>=p)
        {
            int j=p-k+>?p-k+:;
            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("\n");*/
}
int main()
{
    int i,j,te;
    scanf("%d",&te);
    while(te--)
    {
        for(i=;i<;i++)
            scanf("%d",&ans[i]);
        scanf("%s",s);
        int len=strlen(s);
        for(i=;i<len;i++)
        {
            t[len-i-]=s[i];
            if(i==)
                sum[i]=ans[s[i]-'a'];
            else
                sum[i]=sum[i-]+ans[s[i]-'a'];
        }
        t[len]='\0';
        ekmp(s,t,next,extand1);//s做主串
        ekmp(t,s,next,extand2);//t做主串

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

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