HDU 2594 Simpsons’ Hidden Talents （KMP）

標籤：c++   hdu   演算法   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;}