HDU 4300 Contest 1

扩展KMP很容易就明白过来了。

注意的是，后面明文的长度要少于密文，而且当前K+Extend[k]>=L

输出时犯了很多次二，后来人注意吧。

#include <cstdio>
#include <cstring>
#include <string.h>
#include <iostream>
using namespace std;
char t[30];
char table[30];
char code[100005];
char chan[100005];
int next[100005];
int extand[100005];
void Next(const char *T){  
     int len=strlen(T),a=0;  
     next[0]=len;  
     while(a<len-1 && T[a]==T[a+1]) a++;  
     next[1]=a;  
     a=1;  
     for(int k=2;k<len;k++){  
         int p=a+next[a]-1,L=next[k-a];  
         if( (k-1)+L >= p){  
             int j = (p-k+1)>0 ? (p-k+1) : 0;  
             while(k+j<len && T[k+j]==T[j]) j++;  
             next[k]=j;  
             a=k;   
         }   
         else  
             next[k]=L;   
     }   
}

void GetExtand(const char *S,const char *T){   
     int slen=strlen(S),tlen=strlen(T),a=0;   
     int MinLen = slen < tlen ? slen : tlen;  
     while(a<MinLen && S[a]==T[a]) a++;  
     extand[0]=a;  
     a=0;  
     for(int k=1;k<slen;k++){  
         int p=a+extand[a]-1, L=next[k-a];  
         if( (k-1)+L >= p){  
             int j= (p-k+1) > 0 ? (p-k+1) : 0;  
             while(k+j<slen && j<tlen && S[k+j]==T[j]) j++;  
             extand[k]=j;  
             a=k;   
         }  
         else   
             extand[k]=L;
     }   
}
int main(){
 int T,k;
 scanf("%d",&T);
 while(T--){
  cin>>t>>code;
  for(int i=0;i<26;i++){
   table[t[i]-'a']=i+'a';
  }
  int L=strlen(code);
  for(int i=0;i<=L;i++){
   chan[i]=table[code[i]-'a'];
  }
  Next(chan);
  GetExtand(code,chan);
  for(k=0;k<L;k++){
  	if(k+extand[k]>=L&&extand[k]<=k)
  	break;
  }
  for(int i=0;i<k;i++)
  cout<<code[i];
  for(int i=0;i<k;i++)
  cout<<table[code[i]-'a'];
  cout<<endl;
 }
 return 0;
}