HDU 2203(KMP) 親和串

_Phoenix發表於2015-08-24

題意是說迴圈掃描母串A(意思就是可以重複遍歷很多次),如果子串B存在於迴圈的母串之中,那麼輸出yes,否則no。

嗯,注意一點:如果串B的長度大於了串A是不行的,因為按照題意所說的子串是包含在母串中的,所以如果串B的長度超過了串A,輸出應該是no。

 並且很容易就會發現只要兩個母串就能夠表示出所有可能的子串了。

另外,母串ABCD,子串ABCDA,這樣子串長度是超過母串了的,儘管母串重複之後為ABCDABCD包含ABCDA,仍應該輸出no。

程式碼如下:

#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <sstream>
#include <queue>
#include <typeinfo>
#include <fstream>
#include <map>
#include <stack>
typedef long long LL;
using namespace std;
const int MAX = 200002;
int kmp_next[MAX];
void k_next (char *s)
{
	int len = strlen(s);
	kmp_next[0] = -1;
	int k = -1;  //front
	int j = 0;   // behind
	while (j < len)
	{
		if (k == -1 || kmp_next[k] == kmp_next[j])
		{
			++k;
			++j;
			if (kmp_next[k] != kmp_next[j])
				kmp_next[j] = k;
			else
				kmp_next[j] = kmp_next[k];	
		}
		else // shipei
			k = kmp_next[k];
	}
	return;
}
bool k_match(char *s, char *b)
{
	int slen = strlen(s);
	int blen = strlen(b);
	int i = 0;
	int j = 0;
	while (i < slen && j < blen)
	{
		if (j == -1 || s[i] == b[j])
		{
			++i;
			++j;
		}
		else
			j = kmp_next[j];
	}
	if (j == blen)
		return true;
	else
		return false;
}

int main()
{
	char a[MAX], b[MAX];
	char s[MAX];
	while (~scanf("%s %s", a, b))
	{
		strcpy(s, a);
		strcat(s, a);
		if (strlen(a) < strlen(b))
		{
			printf("no\n");
			continue;
		}
		k_next(b);
		if (k_match(s, b))
			printf("yes\n");
		else
			printf("no\n");	
	} 	
	return 0;
} 


相關文章