HDU 2594 Simpsons’ Hidden Talents （KMP）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2594

这题直接用KMP算法就可以做出来，不过我还尝试了用扩展的kmp，这题用扩展的KMP效率没那么高。

KMP算法：

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
int next[50001];
char p[50000],s[50000];
void getnext()
{
int plen=strlen(p),k=0,j=1;
next[0]=-1;next[1]=0;
while (j<plen)
{
if (k==-1||p[j]==p[k])
{
k++;j++;
next[j]=k;
}
else k=next[k];
}
}
int kmp()
{
int i=0,j=0,slen=strlen(s),plen=strlen(p);
while (i<slen)
{
if (j == -1 || s[i] == p[j])
{
i++;
j++;
}
else  j=next[j];
}
if (j==-1) j=0;
return j;
}
int main()
{
int i;
while (scanf("%s%s",p,s)!=EOF)
{
getnext();
i=kmp();
if (i)
{
for (int j=0;j<i;j++)
printf("%c",p[j]);
printf(" %d\n",i);
}
else printf("0\n");
}
return 0;
}

扩展的KMP：

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
char S[50000],T[50000];
int A[50001],B[50001],sl,tl;
void getA()
{
int j=0;
while(1+j<tl&&T[0+j]==T[1+j]) j++;
A[1]=j;
int k=1;
for(int i=2;i<tl;i++)
{
int len=k+A[k]-1,l=A[i-k];
if(l<len-i+1) A[i]=l;
else
{
j=max(0,len-i+1);
while(i+j<tl&&T[i+j]==T[0+j])
j=j+1;
A[i]=j;k=i;
}
}
}
void getB()
{
getA();
int j=0;
while(j<sl&&j<tl&&T[0+j]==S[0+j])
j++;B[0]=j;
int k=0;
for(int i=1;i<sl;i++)
{
int len=k+B[k]-1,l=A[i-k];
if(l<len-i+1) B[i]=l;
else
{
j=max(0,len-i+1);
while(i+j<sl&&j<tl&&S[i+j]==T[0+j])
j=j+1;
B[i]=j,k=i;
}
}
}
int main()
{
int i;
while (~scanf ("%s %s",T,S))
{
int num=0;
sl=strlen(S),tl=strlen(T);
getB();
for (i=0;i<sl;i++)
if (B[i]==sl-i)
{
num=B[i];
break;
}
if (num>0)
{
for (int j=0;j<num;j++)
printf("%c",T[j]);
printf(" ");
}
printf("%d\n",num);
}
return 0;
}