POJ - 1961 Period (KMP next数组求循环节)

题目描述

给定长度为n的字符串， 求串中某个点的前缀中的循环节最多有多少个

样例

Sample Input
3
aaa
12
aabaabaabaab
0

Sample Output
Test case #1
2 2
3 3

Test case #2
2 2
6 2
9 3
12 4

思路
我们知道， KMP中的next数组，next[j] 代表j的前缀和从后面开始， 长度为j的串相等， 我们可以利用这个性质


所以， n 的前缀的循环节的最短为 n - next[n]， 并且需要 n % (n - next[n]) == 0.

代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>

using namespace std;

const int N = 1000010;

int Next[N];
int n;
char s[N];

void get_Next()
{
	Next[1] = 0;
	for(int i = 2, j = 0; i <= n; i++)
	{
		while(j && s[i] != s[j + 1]) j = Next[j];
		if(s[i] == s[j + 1]) j++;
		Next[i] = j;
 	}
}

int main()
{
	int c = 1;
	while(scanf("%d", &n) != EOF && n)
	{
		scanf("%s", s + 1);
		get_Next();
		
		printf("Test case #%d\n", c++);
		for(int i = 2; i <= n; i++)
		{
			int t = i - Next[i];
			if(i % t == 0 && t != i)
				printf("%d %d\n", i, i / t);
		}
		cout << endl;
	}
	
	return 0;
}