hdu 4300

裸exkmp

#include <bits/stdc++.h>
using namespace std;
const int maxn = 100005;
int mp[26];
char str2[maxn],str1[maxn],str3[maxn];
int ne[maxn];
int extend[maxn];
char S[maxn],T[maxn];
void getNext(char *str)
{
    int len=strlen(str),p0,i=0,j;
    ne[0]=len;//初始化next［０］
    while(str[i]==str[i+1]&&i+1<len) i++;
    ne[1]=i;
    p0=1;//初始化p0
    for(i=2;i<len;i++)
    {
        if(ne[i-p0]+i<ne[p0]+p0) ne[i]=ne[i-p0];//第一种情况，可以直接得到next[i]的值
        else//第二种情况，要继续匹配才能得到next[i]的值
        {
            j=ne[p0]+p0-i;//如果i>po+next[po],则要从头开始匹配
            if(j<0) j=0;
            while(i+j<len&&str[i+j]==str[j]) j++;//计算next[i]
            ne[i]=j;
            p0=i;
        }
    }
}
void exkmp(char *str,char *p)//计算extend数组   p 为模式串
{
    int i=0,j,p0,slen=strlen(str),plen=strlen(p);
    getNext(p);//计算p的next数组
    while(i<slen&&i<plen&&str[i]==p[i]) i++;//计算ex[0]
    extend[0]=i;
    p0=0;//初始化po的位置
    for(i=1;i<slen;i++)
    {
        if(ne[i-p0]+i<extend[p0]+p0) extend[i]=ne[i-p0];//第一种情况，直接可以得到ex[i]的值
        else //第二种情况，要继续匹配才能得到ex[i]的值
        {
            j=extend[p0]+p0-i;
            if(j<0) j=0;//如果i>ex[po]+po则要从头开始匹配
            while(i+j<slen&&j<plen&&str[i+j]==p[j]) j++;//计算ex[i]
            extend[i]=j;
            p0=i;//更新po的位置
        }
    }
}
int mp2[26];
int main(){
    int T;
    scanf("%d",&T);
    while(T--){
        scanf("%s%s",str1,str2);
        for( int i = 0;i < 26;i++ ){
            mp[i] = str1[i] - 'a';
            mp2[mp[i]] = i;
        }
        int len =strlen(str2);
        for( int i = 0;i < len;i++ ){
            str3[i] ='a' + mp[str2[i]-'a'];
        }
        exkmp( str3,str2 );
        int res = len;
        for( int i = (len-1)/2+1;i < len;i++ ){
            int dt = min( extend[i],i );
            if( i + dt >= len ) {
                res = i;
                break;
            }
        }
        for( int i = 0;i < res*2;i++ ){
            if( i < res ){
                printf("%c",str2[i]);
            }else{
                printf("%c",'a'+mp2[str2[i-res]-'a']);
            }
        }
        cout << endl;
    }
    return 0;
}