UVa LA 3026 Period

题意

对于每个i，求一个最大整数K>1（如果K存在），使得S的前i个字符组成的前缀是某个字字符串重复K次得到，输出所有存在K的i和对应的K。

分析

这是一个KMP的周期问题，我们知道的是对于字符串来说KMP周期是i - f[i]，那么本题就迎刃而解，注意K > 1，因此，f[i]为0不合法。

代码

#include <stdio.h>      
#include <string.h>  
#include <iostream>  
#include <algorithm>  
      
using namespace std;  
      
const int MAX = 1000010;

int n,f[MAX];
char s[MAX];

void Failure_function(){//模式串的自我匹配
	f[0] = f[1] = 0;
	for(int i=1; i<n; i++){
		int j = f[i];
		while(j && s[i] != s[j]) j = f[j];//失配的话，不断回退
		f[i+1] = (s[i] == s[j]) ? j + 1 : 0;
	}
}

int main(){
	int t = 0;
	while(scanf("%d",&n) == 1 && n){
		scanf("%s",s);
		Failure_function();
		printf("Test case #%d\n",++t);
		for(int i=2; i<=n; i++)
			if(f[i] && i % (i - f[i]) == 0)
				printf("%d %d\n",i, i / (i - f[i]));
		puts("");
	}	
	return 0;
}