HDU 4300 Clairewd’s message（扩展kmp）

Description
给出置换密码的密钥，即第i个字母用哪个字母表示，再给出一个串，串前面为密文，后面为明文，密文一定是完整的，但明文不完整或可能没有，求包含完整密文和完整明文的最短的串
Input
第一行为一整数T(T<=100)表示用例组数，每组用例首先输入26个字符表示密钥，然后属于一个串，串长不超过100000
Output
对于每组用例，输出包含完整密文和完整明文的最短的串
Sample Input
2
abcdefghijklmnopqrstuvwxyz
abcdab
qwertyuiopasdfghjklzxcvbnm
qwertabcde
Sample Output
abcdabcd
qwertabcde
Solution
扩展kmp，由于该串包含完整密文和部分明文，所以该串前半部分必然是密文，可以先将其解密，然后让该串自身做匹配，如果某个后缀的前缀与该串前缀的匹配长度恰好等于这个后缀的长度，即extend[i]=len-i，那么说明i是明文的开始部分，0~i-1就全部是密文
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 222222
char a[maxn],s[maxn],c[33];
int T,nex[maxn];
void extend_kmp(char *a)
{
    int i,j,k,la=strlen(a);
    nex[0]=la;
    for(i=0;i+1<la&&a[i+1]==a[i];i++);
    nex[1]=i;
    k=1;
    for(i=2;i<la;i++) 
    {
        int len=k+nex[k];
        nex[i]=min(nex[i-k],max(0,len-i));
        while(i+nex[i]<la&&a[nex[i]]==a[i+nex[i]])nex[i]++;
        if(i+nex[i]>k+nex[k])k=i;
    }
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s%s",s,a);
        for(int i=0;i<26;i++)
            c[s[i]-'a']=i+'a';
        int len=strlen(a);
        for(int i=0;i<(len+1)/2;i++)a[i]=c[a[i]-'a'];
        extend_kmp(a);
        for(int i=(len+1)/2;i<=len;i++)
            if(nex[i]==len-i)
            {
                for(int k=(len+1)/2;k<i;k++)a[k]=c[a[k]-'a'];
                for(int k=0;k<i;k++)printf("%c",s[a[k]-'a']);
                for(int k=0;k<i;k++)printf("%c",a[k]);
                printf("\n");
                break;
            }
    }
    return 0;
}