hdu2203kmp

http://acm.hdu.edu.cn/showproblem.php?pid=2203
题意 给出主串 模式串 主串可以滚动，即1234->2341，
文能否匹配
可以将主串变为 主串+主串，这样所有的情况就都在内了

#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std;

const int maxn=100005;
char s1[maxn<<1],s2[maxn];
int nex[maxn],flag;

void get_next(int len)
{
    int i=0,j=-1;
    nex[0]=-1;
    while(i<len)
    {
        if(j==-1||s2[i]==s2[j])
        {
            i++;j++;
            nex[i]=j;
        }
        else
        j=nex[j];
    }
}

void kmp(int len1,int len2)
{
    int i=0,j=0;
    get_next(len2);
    while(i<len1)
    {
        if(j==0||s1[i]==s2[j])
        {
            i++;j++;
        }
        else
        j=nex[j];
        if(j==len2)
        {
            flag=1;
            break;
        }
    }
}
int main()
{while(~scanf("%s%s",s1,s2))
 {
     int len1=strlen(s1);
     int len2=strlen(s2);
     if(len1<len2)
     {
         puts("no");
         continue;
     }
     for(int i=0;i<len1;i++)
     s1[len1+i]=s1[i];
     flag=0;
     kmp(len1<<1,len2);
     if(flag)
     puts("yes");
     else
     puts("no");
 }

    return 0;
}