HDU - 4300 Clairewd’s message （拓展kmp）

HDU - 4300 

题意：这个题目好难读懂，，先给你一个字母的转换表，然后给你一个字符串密文+明文，密文一定是全的，但明文不一定是全的，求最短的密文和解密后的明文；

题解：由于密文一定是全的，所以他的长度一定大于整个字符串的一半。将这一半先解密，然后和后缀对比一下，求最长的公共前缀。那么就转换为了拓展kmp问题。

#include<bits/stdc++.h>
using namespace std;

const int maxn=100010;   //字符串长度最大值
int nt[maxn],ex[maxn]; //ex数组即为extend数组
char a[maxn],b[maxn],c[maxn];
map<char,char> mp;

//预处理计算next数组
void GETNEXT(char *str)
{
    int i=0,j,po,len=strlen(str);
    nt[0]=len;    //初始化nt[0]
    while(str[i]==str[i+1]&&i+1<len) i++;    //计算nt[1]
    nt[1]=i;
    po=1;       //初始化po的位置
    for(i=2;i<len;i++){
        if(nt[i-po]+i<nt[po]+po) nt[i]=nt[i-po];    //第一种情况，可以直接得到nt[i]的值
        else{    //第二种情况，要继续匹配才能得到nt[i]的值
            j=nt[po]+po-i;
            if(j<0) j=0;    //如果i>po+nt[po],则要从头开始匹配
            while(i+j<len&&str[j]==str[j+i]) j++;   //计算nt[i]
            nt[i]=j;
            po=i;   //更新po的位置
        }
    }
}

//计算extend数组
void EXKMP(char *s1,char *s2)
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    GETNEXT(s2);    //计算子串的next数组
    while(s1[i]==s2[i]&&i<l2&&i<len) i++;   //计算ex[0]
    ex[0]=i;
    po=0;   //初始化po的位置
    for(i=1;i<len;i++){
        if(nt[i-po]+i<ex[po]+po) ex[i]=nt[i-po];      //第一种情况，直接可以得到ex[i]的值
        else{    //第二种情况，要继续匹配才能得到ex[i]的值
            j=ex[po]+po-i;
            if(j<0) j=0;     //如果i>ex[po]+po则要从头开始匹配
            while(i+j<len&&j<l2&&s1[j+i]==s2[j]) j++;   //计算ex[i]
            ex[i]=j;
            po=i;   //更新po的位置
        }
    }
}

int main()
{
    int t;
    cin>>t;
    while(t--){
        cin>>a>>b;
        int len1=strlen(a);
        int len2=strlen(b);
        for(int i=0;i<len1;i++)
            mp[a[i]]='a'+i;
        for(int i=0;i<len2;i++)
            c[i]=mp[b[i]];
        c[len2]=0;
        EXKMP(b,c);
        int k;
        for(k=0;k<len2;k++)
            if(k+ex[k]>=len2&&k>=ex[k]) break;
        for(int i=0;i<k;i++) cout<<b[i];
        for(int i=0;i<k;i++) cout<<mp[b[i]];
        cout<<endl;
    }
    return 0;
}