POJ - 2406 Power Strings (KMP循环节)

abcd
aaaa
ababab
.

1
4
3

题意：题目要求的是给定一个字符串找到最小循环节的个数，但是这里有个限制的地方就是如果这个字符串不是刚好由n个
最小循环节组成那么就认为一整串才是一个循环节
思路：KMP应用
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1000010;

char pattern[maxn];
int next[maxn];

void getNext() {
	int m = strlen(pattern);
	next[0] = next[1] = 0;
	for (int i = 1; i < m; i++) {
		int j = next[i];
		while (j && pattern[i] != pattern[j])
			j = next[j];
		next[i+1] = pattern[i] == pattern[j] ? j+1 : 0;
	}

	int len = m - next[m];
	if (m % len == 0)
		printf("%d\n", m/len);
	else printf("1\n");
}

int main() {
	while (gets(pattern)) {
		if (strcmp(pattern, ".") == 0)
			break;
		getNext();
	}	
	return 0;
}