【KMP】 HDOJ 2203 亲和串

KMP模板题。。

#include <iostream>  
#include <sstream>  
#include <algorithm>  
#include <vector>  
#include <queue>  
#include <stack>  
#include <map>  
#include <set>  
#include <bitset>  
#include <cstdio>  
#include <cstring>  
#include <cstdlib>  
#include <cmath>  
#include <climits>  
#define maxn 100005
#define eps 1e-6 
#define mod 10007 
#define INF 99999999  
#define lowbit(x) (x&(-x))  
#define lson o<<1, L, mid  
#define rson o<<1 | 1, mid+1, R  
typedef long long LL;  
using namespace std;  

char t[maxn<<1], p[maxn];
int next[maxn];
int n, m;
void init(void)
{
	n=strlen(t);
	m=strlen(p);
	for(int i=0;i<n;i++) t[i+n]=t[i];
	n*=2;
}
void get(void)
{
	next[0]=next[1]=0;
	int i, j;
	for(i=1;i<m;i++){
		j=next[i];
		while(j && p[i]!=p[j]) j=next[j];
		next[i+1]= p[i]==p[j] ? j+1 : 0;
	}
}
void kmp(void)
{
	int i, j;
	for(i=0, j=0;i<n;i++){
		while(j && t[i]!=p[j]) j=next[j];
		if(t[i]==p[j]) j++;
		if(j==m){
			printf("yes\n");
			return;
		}
	}
	printf("no\n");
}
int main(void)
{
	while(scanf("%s%s", t, p)!=EOF){
		init();
		get();
		kmp();
	}
	return 0;
}