hdu4300---Clairewd’s message(扩展KMP)

题意
给你一个26个字母的转换表,把密文转换成明文的转换表.
举个例子:
比如转换表是qwertyuiopasdfghjklzxcvbnm
对应的
密文q的明文是a
密文w的明文是b
密文e的明文是e
…
刚开始的时候一直没搞明白这个转换表.
然后再给你一个字符串,字符串是由密文+明文构成的,密文是完整的,明文是有可能缺失一部分的.前面的密文翻译过来以后就是后面的明文.
第一个字符串是转换表,第二个字符串是密文+明文的字符串,我们的目标是找出密文和明文的分界线,然后补全后面的明文,(前面的完整的密文就是后面明文的内容),所以这题的主要关键就是求明文与密文的分界线.
思路
我们如果把密文+明文的字符串全部破解,那就会变成明文+乱码,和原来的密文+明文对比来看,其实就是看这两个字符串的前缀和后缀,这刚好可以用扩展KMP来写,我们求出extend数组以后,我们从前往后判断,第一个满足i+extend[i]>=n,而且i>=extend[i]的就是分界线,剩下的事情就很简单了.
代码

#include<stdio.h>
#include<string.h>
#include<map>
#include<algorithm>
using namespace std;
const int N = 10005;
int next[N],extend[N];
char tab[N],str[N],p[N];
map<char,char>mp;
void getNext(char p[]){
	int len=strlen(p);
	int i=0;
	next[0]=len;
	while(p[i]==p[i+1]&&i+1<len) i++;
	next[1]=i;
	int po=1;
	for(int i=2;i<len;i++){
		if(next[i-po]+i<next[po]+po) 
		       next[i]=next[i-po];
		else{
			int j=next[po]+po-i;
			if(j<0) j=0;
			while(i+j<len&&p[j]==p[i+j])
			   j++;
			next[i]=j;
			po=i;
		}
	}
}
void EKMP(char str[],char p[]){
	int len=strlen(str);
	int m=strlen(p);
	getNext(p);
	int i=0;
	while(str[i]==p[i]&&i<m&&i<len)
	   i++;
	extend[0]=i;
	int po=0;
	for(int i=1;i<len;i++){
		if(next[i-po]+i<extend[po]+po) 
		   extend[i]=next[i-po];
		else{
			int j=extend[po]+po-i;
			if(j<0) j=0;
			while(i+j<len&&j<m&&str[j+i]==p[j])
			    j++;
			extend[i]=j;
			po=i;
 		}
	}
}
int main(){
	int T;
	scanf("%d",&T);
	while(T--){
		memset(extend,0,sizeof(extend));
		memset(next,0,sizeof(next));
		scanf("%s%s",tab,str);
		int n=strlen(tab);
		int m=strlen(str);
		for(int i=0;i<n;i++)
		    mp[tab[i]]='a'+i;
		for(int i=0;i<m;i++)
		   p[i]=mp[str[i]];
		getNext(p);
		EKMP(str,p);
		int k;
		for(k=0;k<m;k++){
			if(k+extend[k]>=m&&k>=extend[k])
			  break;
		}
		for(int i=0;i<k;i++) printf("%c",str[i]);
		for(int i=0;i<k;i++) printf("%c",mp[str[i]]);
		printf("\n");
	}
	return 0;
}