hdu 4300 Clairewd’s message (扩展KMP)

这道题问的就是将1个串如何变为stringA+stringB的形式，使得stringA是stringB经过映射得到相同的串。映射那步其实没有什么 价值，假设str为原串s经过映射后得到的串，我们可以以str为模式串，以s为原串做一次扩展KMP，得到extend数组，extend[i]表示原 串以第i开始与模式串的前缀的最长匹配。经过O(n)的枚举，我们可以得到,若extend[i]+i=len且i>=extend[i]时，表示 stringB即为该点之前的串，stringA即为该点之前的str串，最后输出即可。


#include<stdio.h>
#include<string.h>
#include<map>
#define maxn 200000
using namespace std;
map<char,char>map1;
int max(int x,int y)
{
    return x>y?x:y;
}
char s[maxn],tab[maxn],c[maxn];
int next[maxn],extend[maxn];
void EKMP(char s[],char t[])//s[]为主串，t[]为模版串
{
    int i,j,p,l;
    int len=strlen(t);
    int len1=strlen(s);
    memset(next,0,sizeof(next));
    memset(extend,0,sizeof(extend));

    next[0]=len;
     j=0;
    while(1+j<len&&t[j]==t[1+j])j++;

    next[1]=j;


    int a=1;
    for(i=2;i<len;i++)
    {
        p=next[a]+a-1;
        l=next[i-a];
        if(i+l<p+1)next[i]=l;
        else
        {
             j=max(0,p-i+1);
            while(i+j<len&&t[i+j]==t[0+j])j++;

            next[i]=j;
            a=i;
        }
    }



    j=0;
    while(j<len1&&j<len&&s[j]==t[j])j++;

    extend[0]=j;

    

    a=0;

    for(i=1;i<len1;i++)
    {
        p=extend[a]+a-1;
        l=next[i-a];
        if(l+i<p+1)extend[i]=next[i-a];
        else
        {
            j=max(0,p-i+1);
            while(i+j<len1&&j<len&&s[i+j]==t[j])j++;

            extend[i]=j;
            a=i;
        }
    }


}
int main()
{
    int i,j;
    int t;
    scanf("%d",&t);
    while(t--)
    {


        scanf("%s%s",tab,s);
        int len=strlen(tab);
        int len1=strlen(s);

        for(i=0;i<len;i++)
        {
            map1[tab[i]]='a'+i;
        }
        for(i=0;i<len1;i++)
        {
            c[i]=map1[s[i]];
        }
        c[i]='\0';
        EKMP(s,c);

        for(i=0;i<len1;i++)
        {
            if(i+extend[i]>=len1&&i>=extend[i])
            {

                break;

            }
        }
       
        for(j=0;j<i;j++)printf("%c",s[j]);
        for(j=0;j<i;j++)printf("%c",map1[s[j]]);
        printf("\n");


    }
}