Simpsons’ Hidden Talents HDU - 2594

原题链接

KMP+字符串的运用

一开始计算错了时间复杂度,还以为可以直接暴力....

错解思路：

       求公共最长的前缀和后缀,如果直接将字符串拼接在一起,如果遇到这样的测试样例eg:s1 = aaa,s2 = aaa就会输出错误,所以我们必须分隔开s1和s2

正解思路:

        我们不能让s2认为属于s1部分的也是它自己的,所以我们可以在拼接s1和s2的时候插入一个空格,这样就可以分隔开s1和s2

易错点：

        如果我们从后缀开始求匹配的字符串就会遇到如下错误

#include <bits/stdc++.h>
using namespace std;
const int N = 50010;//暴力会TLE,不要计算错时间复杂度 
string s1,s2;
char a[N*2];
int ne[N*2];
int main()
{
    while(cin>>s1)
    {
        int idx = 0; bool flag = true;
        cin>>s2;
        string prime = s1+" "+s2;
        int alen = s1.size(); int blen = s2.size();
        while(idx<prime.size()){
            a[idx+1] = prime[idx];
            idx++;
        }
        a[idx+1] = '\0';
        int clen = strlen(a+1);
        for(int i=2,j=0;i<=clen;i++){
            while(j&&a[i]!=a[j+1]) j = ne[j];
            if(a[i]==a[j+1]) j++;
            ne[i] = j;
        }
        if(ne[clen]<=alen){
            int st = clen-ne[clen]+1;
            if(ne[st]==1){
                flag = false;
                for(int i=st;i<=clen;i++) printf("%c",a[i]);
                printf(" %d\n",ne[clen]);
            }
        }
        if(flag) printf("0\n");
    }
    return 0;
} 

我们不需要判断匹配的字符串起始是否匹配,因为如果最后一个字符ne[i]!=0,那么前面肯定匹配

以下是修正代码

#include <bits/stdc++.h>
using namespace std;
const int N = 50010;//暴力会TLE,不要计算错时间复杂度 
string s1,s2;
char a[N*2];
int ne[N*2];
int main()
{
    while(cin>>s1)
    {
        int idx = 0; bool flag = true;
        cin>>s2;
        string prime = s1+" "+s2;
        int alen = s1.size(); int blen = s2.size();
        while(idx<prime.size()){
            a[idx+1] = prime[idx];
            idx++;
        }
        a[idx+1] = '\0';
        int clen = strlen(a+1);
        for(int i=2,j=0;i<=clen;i++){
            while(j&&a[i]!=a[j+1]) j = ne[j];
            if(a[i]==a[j+1]) j++;
            ne[i] = j;
        }
        if(ne[clen]<=alen&&ne[clen]){
            for(int i=1;i<=ne[clen];i++) printf("%c",a[i]);
            printf(" %d\n",ne[clen]);
            flag = false;
        }
        if(flag) printf("0\n");
    }
    return 0;
}