hdu 1867 A + B for you again(KMP)

最新推荐文章于 2021-07-22 15:35:32 发布

huanzhizun

最新推荐文章于 2021-07-22 15:35:32 发布

阅读量589

点赞数

分类专栏： KMP 文章标签： ACM kmp

本文链接：https://blog.csdn.net/u013665921/article/details/26340257

版权

KMP 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

简单KMP，但后缀数组超时。

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<string>
using namespace std;
int nex1[200005],nex2[200005];
char a[100005],b[100005],f[200005],c[200005],d[200005];
void kmp1(int n,int m)
{
    int i,j,k,t;
    k=0;
    t=-1;
    nex1[0]=-1;
    while(k<n)
    {
        if(t==-1||a[k]==a[t])
        {
            k++;
            t++;
            nex1[k]=t;
        }
        else
            t=nex1[t];
    }
}
void kmp2(int n,int m)
{
    int k,t;
    k=0;
    t=-1;
    nex2[0]=-1;
    while(k<n)
    {
        if(t==-1||b[k]==b[t])
        {
            k++;
            t++;
            nex2[k]=t;
        }
        else
            t=nex2[t];
    }
}
int nn1(int n,int m)
{
    int p,k;
    p=0;
    k=0;
    while(p<n)
    {
       if(k==-1||a[p]==b[k])
       {
           p++;
           k++;
       }
        else
            k=nex2[k];
    }
    return k;
}
int nn2(int n,int m)
{
    int p,k;
    p=0;
    k=0;
    while(p<m)
    {
        if(k==-1||b[p]==a[k])
        {
            p++;
            k++;
        }
        else
            k=nex1[k];
    }
    return k;
}
int main()
{
    int i,n,m,p,k;
    while(cin>>a>>b)
    {
        n=(int)strlen(a);
        m=(int)strlen(b);
        kmp1(n,m);
        kmp2(n,m);
        p=nn1(n,m);
        k=nn2(n,m);
        if(p<k)
        {
            for(i=0;i<m;i++)
                printf("%c",b[i]);
            for(i=k;i<n;i++)
                printf("%c",a[i]);
        }
        else if(p>k)
        {
            for(i=0;i<n;i++)
                printf("%c",a[i]);
            for(i=p;i<m;i++)
                printf("%c",b[i]);
        }
        else
        {
            for(i=0;i<m;i++)
                c[i]=b[i];
            for(i=m;i<n+m-k;i++)
                c[i]=a[i-m+k];
            c[n+m-k]='\0';
            for(i=0;i<n;i++)
            {
                d[i]=a[i];
            }
            for(i=n;i<n+m-p;i++)
                d[i]=b[i-n+p];
            d[n+m-p]='\0';
            if(strcmp(a,b)<0)
                printf("%s",d);
            else
                printf("%s",c);
                
        }
        printf("\n");
    }
    return 0;
}