HDU-2203 kmp水一发

#include <cstdio>
#include <cstring>
#include <vector>
#include <queue>
#include <iostream>
#include <cmath>
#include <vector>
#include <algorithm>
#include <map>
#include <set>
using namespace std;
typedef long long ll;
#define DEBUG
const int maxn=110000+5,maxv=26,INF=0x3f3f3f3f,mod=100000000;
int _next[maxn];
char str[maxn],pat[maxn];
void getnext(char* pat){
    int n=strlen(pat);
    int i=0,j=0;
    for(int i=1;i<n;i++){
        j=_next[i];
        while(j&&pat[j]!=pat[i])j=_next[j];
        _next[i+1]=pat[i]==pat[j]?j+1:0;
    }
}
void kmp(char* str,char* pat){
    int n1=strlen(str),n2=strlen(pat);
    int i=0,j=0;
    while(i<n1){
        while(str[i]!='\0'&&str[i]==pat[j])i++,j++;
        if(j==n2){
            // printf("ok %d\n",i-j);
            printf("yes\n");
            return;
        }
        while(j&&str[i]!=pat[j])j=_next[j];
        if(str[i]==pat[j])j++;
        if(j==n2){
            // printf("ok %d\n",i-j+1);
            printf("yes\n");
            return;
        }
        i++;

    }
    printf("no\n");
    return;
}
int main(){
#ifdef DEBUG
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    while(cin>>str>>pat){
        int n=strlen(str);
        int n2=strlen(pat);
        for(int i=0;i<n2-1;i++){
            str[n+i]=str[i];
        }
        str[n+n2-1]='\0';
        memset(_next,0,sizeof(_next));
        getnext(pat);
        // for(int i=0;i<strlen(pat);i++){
        //  printf("%d ",_next[i]);
        // }
        kmp(str,pat);
    }
#ifdef DEBUG
    fclose(stdin);
    fclose(stdout);
#endif
    return 0;
}