ekmp

ekmp

http://acm.hdu.edu.cn/showproblem.php?pid=4300

暴力
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
 
const int MAXN = 1000005;
char a[50];
char b[MAXN];
char str[MAXN];
int  f[MAXN];
int  val[200];
 
int main(){
    int nCase;
    scanf("%d",&nCase);
    while(nCase--){
        scanf("%s %s",a,b);
        for(int i=0; a[i]; ++i){
            val[a[i]] = i+'a';
        }
        int len=strlen(b);
        int mid;
        if(len & 1) mid=len/2;
        else mid=len/2-1;
        int pos;
        for(int i=mid; i<len; ++i){
            bool flag=true;
            for(int k=0,j=i+1; j<len; ++j,++k){
                if(val[b[k]]!=b[j]){
                    flag=false;
                    break; 
                }
            }
            if(flag){
                pos=i;
                break;    
            }
        }
        memset(str, 0, sizeof(str));
        for(int i=0; i<=pos; ++i)
            str[i]=b[i];
        for(int i=pos+1,k=0; k<=pos; ++i,++k)
            str[i]=val[str[k]];
        puts(str);
    }
    return 0;
}

ekmp

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
 
typedef long long int64;
const int MAXN = 1000005;
char a[50];
char b[MAXN];
char str[MAXN];
int  next[MAXN];
int  extend[MAXN];
int  hash[200];
 
void getNext(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 EKMP(const char* S,const char* T){
    getNext(T);
    int slen=strlen(S), tlen=strlen(T), a=0;
    int minLen = min(slen,tlen);
    while(a<minLen && S[a]==T[a]) ++a;
    extend[0] = a;
    a=0;
    for(int k=1; k<slen; ++k){
        int p=a+extend[a]-1, L=next[k-a];
        if(k-1+L >= p){
            int j=max(p-k+1, 0);
            while(k+j<slen && j<tlen && S[k+j]==T[j]) ++j;
            extend[k] = j;
            a=k;
        }
        else
            extend[k] = L;
    }
} 
 
int main(){
    int nCase;
    scanf("%d",&nCase);
    while(nCase--){
        scanf("%s %s",a,b);
        for(int i=0; a[i]; ++i){
            hash[a[i]] = i+'a';
        }
        memset(str, 0, sizeof(str));
        for(int i=0; b[i]; ++i){
            str[i] = hash[b[i]];
        }
        EKMP(b, str);
        int len=strlen(b);
        int maxi=len;
        for(int i=0; i<len; ++i){
            if(i+extend[i]>=len && i>=extend[i]){
                maxi=i;
                break;
            }
        }
        memset(str, 0, sizeof(str));
        for(int i=0; i<maxi; ++i){
            str[i]=b[i];
            str[maxi+i]=hash[b[i]];
        }
        puts(str);
    }
    return 0;
}