HDU 4300 Clairewd’s message(EXKMP)

题意：给你一个二十六位的密码表A，然后给你一段文字S。前面部分完整的是密文，
后面是未必完整的明文，最后输出完整的密文+完整的明文（长度要求最短）
思路：
首先将文字按照密码表转换
S中的密文部分就全部都变成了明文，而明文部分都变成了乱七八糟snaosjaoixwqaw（用不到）。保存为T
所以 不完整的明文部分可以通过 S的后缀和T的最长公共前缀（EXKMP模板）得出，求出的不完整明文保证最长，这样就可以让密文最短，知道了不完整明文的长度，也就可以知道了完整密文长度

#include<iostream>
#include<algorithm>
#include<string.h>
#include<map>
#include<queue>
#include<cmath>
#include<cstdio>
#include<stack>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=1e5+5;
const int MOD=1e9+7;
int nex[maxn];
int extend[maxn];
void pre_exkmp(char *b)
{
    int len=strlen(b);
    nex[0]=len;
    int j=0;
    while(j+1<len&&b[j]==b[j+1]) j++;
    nex[1]=j;
    int k=1;
    for(int i=2;i<len;i++)
    {
        int p=nex[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) nex[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<len&&b[i+j]==b[j]) j++;
            nex[i]=j;
            k=i;
        }
    }
}
void exkmp(char *a,char *b)
{
    pre_exkmp(b);
    int j=0;
    int lena=strlen(a);
    int lenb=strlen(b);
    while(j<lena&&j<lenb&&a[j]==b[j]) j++;
    extend[0]=j;
    int k=0;
    for(int i=1;i<lena;i++)
    {
        int p=extend[k]+k-1;
        int L=nex[i-k];
        if(i+L<p+1) extend[i]=L;
        else
        {
            j=max(0,p-i+1);
            while(i+j<lena&&j<lenb&&a[i+j]==b[j]) j++;
            extend[i]=j;
            k=i;
        }
    }
}
char a[maxn];//主串
char b[maxn];
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		char code[30];
		scanf("%s%s",code,a);
		map<char,char> mp;
		for(int i=0;i<26;i++)
			mp[code[i]]='a'+i;
		int len=strlen(a);
		for(int i=0;i<len;i++)
			b[i]=mp[a[i]];
		b[len]='\0';
		exkmp(a,b);
		int i;
		for(i=0;i<len;i++)
			if(len-i==extend[i]&&i>=extend[i])//后缀与最长公共前缀完全相同 并且 i要大于最长公共前缀长度（密文要大于不完整明文） 
				break;
		for(int j=0;j<i;j++)
			printf("%c",a[j]);
		for(int j=0;j<i;j++)
			printf("%c",mp[a[j]]);
		printf("\n");
	}
	return 0;
}