HDU - 4300 多校A题 Clairewd’s message

题目链接：https://cn.vjudge.net/contest/226030#problem/A

题目首先给出译码本，译码本给出了26字母的映射
然后再给出一段，截获的文本y，这段文本y的组成是：完整的密文+一部分明文。
所以我们可以把这段文本按照译码本先译码成一个文本x，文本x的组成：完整的明文 + 一串字符
这样一来我们就可以看得出，这就是一个裸的扩展kmp算法：寻找y[i…n-1]与x[0…m-1]的最长公共前缀。
怎么样才算找到最短的完整文本呢，首先由于密文是完整的，所以明文的开头肯定在文本y的后半段，所以我们可以从文本y的中间开始枚举明文的开头，直到找到一个 y[i…n-1] 与 x[0…m-1]的最长公共前缀长度等于len - i 即可，这里的len就是文本y的长度。
代码如下：

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <string>
#include <stack>

using namespace std;

typedef long long LL;
typedef unsigned long long uLL;

//裸的拓展kmp，求最长公共前缀，这里用的是kuangbin大佬的板子

const int maxn = 200000 + 10;

void pre_EKMP(char x[], int m, int next[]){
    next[0] = m;
    int j = 0;
    while(j+1<m && x[j]==x[j+1]) j++;
    next[1] = j;
    int k = 1;
    for(int i=2; i<m; i++)
    {
        int p = next[k]+k-1;
        int L = next[i-k];
        if(i+L < p+1) next[i] = L;
        else
        {
            j = max(0, p-i+1);
            while(i+j<m && x[i+j]==x[j]) j++;
            next[i] = j;
            k = i;
        }
    }
}

void EKMP(char x[], int m, char y[], int n, int next[], int extend[])
{
    pre_EKMP(x, m, next);
    int j = 0;
    while(j<n && j<m && x[j]==y[j]) j++;
    extend[0] = j;
    int k = 0;
    for(int i = 1; i < n; i++)
    {
        int p = extend[k] + k - 1;
        int L = next[i-k];
        if(i+L < p+1) extend[i] = L;
        else
        {
            j = max(0, p-i+1);
            while(i+j<n && j<m && y[i+j] == x[j]) j++;
            extend[i] = j;
            k = i;
        }
    }
}

int main(){
    int T;
    scanf("%d", &T);
    while(T--){
        char change[300], x[maxn], y[maxn];
        for(char i = 'a'; i <= 'z'; ++i) {
            char ch;
            scanf(" %c", &ch);
            change[ch] = i;
        }

        scanf("%s", x);
        int len = strlen(x);
        for(int i = 0; i < len; ++i)
            y[i] = change[x[i]];

        int extend[maxn], next[maxn], i;
        EKMP(y, strlen(y), x, strlen(x), next, extend);
        bool flag = false;
        for(i = (len-1)/2+1; i < len; ++i){
            if(extend[i] == len-i){
                flag = true;
                break;
            }
        }
        if(flag || i == len){
            for(int j = 0; j < i; ++j)
                putchar(x[j]);
            for(int j = 0; j < i; ++j)
                putchar(y[j]);
            putchar('\n');
        }
    }
    return 0;
}