hdu--2594 Simpsons’ Hidden Talents

题目地址

hdu2594’

题目大意

就是求s1的前缀和s2的后缀的最长公共长度。

解题思路

拓展kmp不知道的可以看一下这里。
拓展kmp的定义是求字符串T(长度为m)和字符串S(长度为n)的每一个后缀的最长公共前缀。那么对于题目输入的s1和s2就需要换一下位置，s1为T, s2为S。那可能有人就有疑惑了，这不是要s1的前缀嘛，傻孩子，你求出来的extend[i]数组，代表的是T和S[i, n-1]最长公共前缀长度，里面一定是前缀，因为字符串本身也是一个前缀。

需要注意的是，在extend中查找最大值时，需要判断一下extend[i] + i是否等于strlen(s2)，为什么要等于母串长度呢，因为题目要的是s2的后缀和s1的前缀的最长公共长度，extend数组里面是s2的后缀和s1的前缀的最长公共前缀，不是整个s2后缀。通过判断就能保证是整个s2后缀。

AC代码

#include <iostream>
#include <cstring> 

using namespace std;

const int MAX=100010; //字符串长度最大值
int Next[MAX],extend[MAX];

//预处理计算Next数组
void getNext(char str[])
{
    int i=0,j,po,len=strlen(str);
    Next[0]=len; //初始化next[0]
    while(str[i]==str[i+1] && i+1<len) i++; Next[1]=i; //计算next[1]
    po=1; //初始化po的位置
    for(i=2;i<len;i++)
    {
        if(Next[i-po]+i < Next[po]+po) //第一种情况，可以直接得到next[i]的值
            Next[i]=Next[i-po];
        else //第二种情况，要继续匹配才能得到next[i]的值
        {
            j = Next[po]+po-i;
            if(j<0) j=0; //如果i>po+next[po],则要从头开始匹配
            while(i+j<len && str[j]==str[j+i]) j++; Next[i]=j;
            po=i; //更新po的位置
        }
    }
}

//计算extend数组
void EXKMP(char s1[],char s2[])
{
    int i=0,j,po,len=strlen(s1),l2=strlen(s2);
    getNext(s2); //计算子串的next数组
    while(s1[i]==s2[i] && i<l2 && i<len) i++; extend[0]=i;
    po=0; //初始化po的位置
    for(i=1;i<len;i++)
    {
        if(Next[i-po]+i < extend[po]+po) //第一种情况，直接可以得到extend[i]的值
            extend[i]=Next[i-po];
        else //第二种情况，要继续匹配才能得到extend[i]的值
        {
            j = extend[po]+po-i;
            if(j<0) j=0; //如果i>extend[po]+po则要从头开始匹配
            while(i+j<len && j<l2 && s1[j+i]==s2[j]) j++; extend[i]=j;
            po=i; //更新po的位置
        }
    }
}

int main()
{
    char s[50005], t[50005];
    
    while (cin >> s >> t)
    {
        int len = strlen(t);
        int maxx = 0;
        
        EXKMP(t, s);
        
        for (int i=0; i<len; i++)
        {
            if (extend[i]>maxx && extend[i]+i==len)
            {
                maxx = extend[i];
            }
        }
        
        if (maxx)
        {
            for (int i=0; i<maxx; i++)
            {
                cout << s[i];
            }
            
            cout << " " << maxx << endl;
        }
        else
        {
            cout << 0 << endl;
        }
    }
    
    return 0;
}