HDU 1867 A + B for you again(KMP:后缀与前缀)

HDU 1867 A + B for you again(KMP:后缀与前缀)

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

题意:

        给你两个字符串,输出他们合并之后的字符串,合并的时候把A的后缀和B的前缀重叠合(或者把A的前缀和B的后缀重合)。要求合并后的串既包含A右包含B，且使得合并后的字符串尽量短，其次是使得合并后的字符串字典序尽量小.

分析:

       首先A和B合并他们一定是首尾重叠相连，要求合并后字典序最小，所以当合并后串长度一样时，我们要把A或B中字典序小的放在前面。

       然后计算A的后缀和B的前缀匹配长度为len1，计算A的前缀和B的后缀匹配长度为len2。

       如果len1大，那么就把A放前面。如果len2大，那么就把B放前面。

       此题可以暴力直接判断A与B相同的部分,该方法更快.不过为了熟练KMP算法,我还是用KMP写吧。

AC代码:

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
const int MAXN=100000+1000;
char *S,*T;
char str1[MAXN],str2[MAXN];
int next[MAXN];
int n,m;
void getFail(char *T)
{
    next[0]=next[1]=0;
    for(int i=1;i<m;i++)
    {
        int j=next[i];
        while(j && T[i]!=T[j]) j=next[j];
        next[i+1] = (T[i]==T[j])?j+1:0;
    }
}
int KMP(char *S,char *T)
{
    int j=0;
    n=strlen(S);
    m=strlen(T);
    getFail(T);
    for(int i=0;i<n;i++)
    {
        while(j && S[i]!=T[j]) j=next[j];
        if(S[i]==T[j]) j++;
        if(i==n-1) return j;
    }
}
int main()
{
    while(scanf("%s%s",str1,str2)==2)
    {
        int len1=KMP(str1,str2);
        int len2=KMP(str2,str1);
        if(len1==len2)
        {
            if(strcmp(str1,str2)<0)
            {
                printf("%s",str1);
                printf("%s\n",str2+len1);
            }
            else
            {
                printf("%s",str2);
                printf("%s\n",str1+len2);
            }
        }
        else if(len1<len2)
        {
            printf("%s",str2);
            printf("%s\n",str1+len2);
        }
        else
        {
            printf("%s",str1);
            printf("%s\n",str2+len1);
        }
    }
    return 0;
}