1. 程式人生 > >HDU 4300 Clairewd’s message(拓展KMP)

HDU 4300 Clairewd’s message(拓展KMP)

題意
這道題的題意很奇怪。。給你兩個字串,第一個字串 s h sh 表示的是密碼錶,長度為26,其中 s h [ 0

] sh[0] 表示的是‘a’加密過後的字元, s h [ 1 ] sh[1]
表示的是’b’加密後的字元等等。。第二個字元表示的是給你一段字元他是由加密過後的字元和一段未加密的字元,其中未加密的字元不完整,前半段加密的字元是完整的,並且在完整的情況下加密字元和未加密字元應該是相同的,現在想讓你恢復這整個字串且讓你新增的字母最少
思路
首先字串由密文+明文組成,且密文是完整的,我們將第二個串s整個都解密記為p,那麼p的所有字首都是解密後的,而s的所有後綴也都是解密後的,那麼顯然我們是要在s的所有後綴裡面去找p的字首,且這個字尾一定要超過s的串長,要在本來串的中點之後,如果找到了就說明有,如果沒有找到就說明一個明文都沒有,直接輸出s串的解密文就好了
程式碼

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100000 + 1000;
string ch , sh , ph;
int nex[maxn] , ex[maxn];
map<char,char>M;
void pre_kmp()
{
	memset(nex,0,sizeof(nex));
	int len = ph.size();
	int j = 0, k = 1;
	nex[0] = len;
	while(j+1 < len && ph[j] == ph[j+1]) j++;
	nex[1] = j;
	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 && ph[j+i] == ph[j]) j ++;
			k = i;nex[i] = j;
		}
	}
}
void exkmp()
{
	memset(ex,0,sizeof(ex));
	int lenp = ph.size() , lens = sh.size();
	int j = 0 , k = 0;
	while(j < lenp && j < lens && ph[j] == sh[j]) j++;
	ex[0] = j;
	for(int i = 1 ; i < lenp ; i++)
	{
		int p = ex[k] + k - 1;
		int L = nex[i-k];
		if(i + L < p + 1) nex[i] = L;
		else 
		{
			int j = max(0,p-i+1);
			while(i+j<lens && j < lenp && ph[j] == sh[i+j]) j++;
			k = i ; ex[i] = j;
		}
	}
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		cin>>ch>>sh;
		ph.clear();
		M.clear();
		for(int i = 0 ; i < ch.size() ; i ++) M[ch[i]] = (i+'a');
		for(int i = 0 ; i < sh.size() ; i ++) ph += M[sh[i]];
		pre_kmp();
		exkmp();
		int pos = sh.size();
		for(int i = 0 ; i < sh.size() ; i++)
		{
			if(i + ex[i] >= sh.size() && i >= ex[i])
			{
				pos = i;break;
			}
		}
		for(int i = 0 ; i < pos ; i++) cout<<sh[i];
		for(int i = 0 ; i < pos ; i++) cout<<M[sh[i]];puts("");
	}
	return 0;
	
}