Clairewd’s message HDU - 4300(阅读理解+扩展kmp)

Clairewd’s message HDU - 4300

题意:

给你两个字符串p,s.
p代表从a到z26个英文字母所对应的密码
s代表你获得的密文+明文（明文有可能不全）
让你输出最短的完整s（有可能需要补全）
题意纯属阅读理解^ ^

---

解题思路：

题意杀我
扩展kmp。我们求得了s中，每个位置的z函数后。进行判断。

1. i+z[i]-1<len-1舍去 我们从i点开始向后走z[i]，还有剩余的，不满足题意，舍去
2. z[i]-1≥i 舍去 我们从头开始走z[i]会超过i,z[i]过大，不满足题意，舍去

---

AC代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
//#pragma GCC optimize(3,"Ofast","inline")
#define LL long long
#define MT(a,b) memset(a,b,sizeof(a))
const int mod=1000007;
const int maxn=2e5+5;
const int ONF=-0x3f3f3f3f;
const int INF=0x3f3f3f3f;
int z[maxn];
char s[maxn],password[30],repassword[300];
void get_password(){
    for (int i=0;i<26;i++){
        repassword[password[i]]='a'+i;
    }
}
void get_z(char *str){
    int len=strlen(str);
    MT(z,0);
    for (int i=1,l=0,r=0;i<len;i++){
        if (i<=r) z[i]=min(r-i+1,z[i-l]);
        while (str[i+z[i]]==repassword[str[z[i]]]&&(i+z[i]-1<len)) z[i]++;
        if (i+z[i]-1>=r) l=i,r=i+z[i]-1;
    }
}
void solve(char *str){
    get_password();
    get_z(str);
    int len=strlen(str);
    int ans=len;
    for (int i=0;i<len;i++){
        if (i+z[i]-1!=len-1) continue;
        if (z[i]-1>=i) continue;
        ans=min(ans,i);
    }
    for (int i=0;i<ans;i++) printf("%c",str[i]);
    for (int i=0;i<ans;i++) printf("%c",repassword[str[i]]);
    printf("\n");
}
int main (){
    int T;cin>>T;while (T--){
        scanf("%s%s",password,s);
        solve(s);
    }
    return 0;
}