[算法竞赛进阶指南] 周期 (KMP)

题目

一个字符串的前缀是从第一个字符开始的连续若干个字符，例如”abaab”共有5个前缀，分别是a, ab, aba, abaa, abaab。

我们希望知道一个N位字符串S的前缀是否具有循环节。

换言之，对于每一个从头开始的长度为 i （i>1）的前缀，是否由重复出现的子串A组成，即 AAA…A （A重复出现K次,K>1）。

如果存在，请找出最短的循环节对应的K值（也就是这个前缀串的所有可能重复节中，最大的K值）。

输入格式

输入包括多组测试数据，每组测试数据包括两行。

第一行输入字符串S的长度N。

第二行输入字符串S。

输入数据以只包括一个0的行作为结尾。

输出格式

对于每组测试数据，第一行输出 “Test case #” 和测试数据的编号。

接下来的每一行，输出具有循环节的前缀的长度i和其对应K，中间用一个空格隔开。

前缀长度需要升序排列。

在每组测试数据的最后输出一个空行。

数据范围

2≤N≤1000000

输入样例：

3
aaa
4
abcd
12
aabaabaabaab
0

输出样例：

Test case #1
2 2
3 3

Test case #2

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

分析：

• 这个题考察的是对KMP算法中next数组的理解。
• next数组的意思就是当前位置前面的最大相同前后缀，我们运用这一点可以解决该题。

代码

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N=1000000;

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

void get_next(){ //获得next数组 
	Next[0]=-1;
	int k=-1;
	int j=0;
	while(j<n){
		if(k==-1||str[j]==str[k]){
			k++;
			j++;
			Next[j]=k; 
		}else{
			k=Next[k];
		}
	}
}

int main()
{
	int cas=1;
	while(cin>>n&&n){
		memset(Next,0,sizeof(Next));
		cin>>str;
		get_next();
		cout<<"Test case #"<<cas++<<endl;
		for(int i=1;i<=n;i++){
			int t=i-Next[i];
			if(t!=i&&i%t==0)
			cout<<i<<" "<<i/t<<endl; 
		}
		cout<<endl;
	} 
	return 0;
}