HDU2203

#include<stdio.h>
#include<string.h>
#define N 100005
int next[N];
int flag,len2,len1;
char str1[2*N],str2[N];

void getnext()
{
    int i,j,k;
    memset(next,0,sizeof(next));
    next[0]=-1;
    j=0;
    k=-1;
    while(j<len2)
    {
        if(k==-1||str2[j]==str2[k])
        {
            k++;
            j++;
            next[j]=k;
        }
        else
            k=next[k];
    }
    return ;
}

void kmp()
{
    int i,j,k;
    getnext();
    j=k=0;
    while(j<len1&&k<len2)
    {
        if(k==-1||str1[j]==str2[k])
        {
            j++;
            k++;
        }
        else
            k=next[k];
    }
    if(k==len2)
    {
        flag=1;
    }
    return ;
}

int main()
{
    int i,j,k;
    while(scanf("%s%s",str1,str2)!=EOF)
    {
        flag=-1;
        len1=strlen(str1);
        len2=strlen(str2);
        if(len1<len2)
        {
            printf("no\n");
            continue;
        }
        for(i=0;i<len1;i++)
            str1[i+len1]=str1[i];
        str1[len1*2]='\0';
        len1=len1*2;
        kmp();
        if(flag==1) printf("yes\n");
        else printf("no\n");
    }
    return 0;
}

关键在于循环。

可以把str1字符串再接在str1字符串的后面，这样所有的循环都考虑到了。

不过这样要在kmp之前，判断len1，len2.