1. 程式人生 > >POJ 2406 Power Strings(KMP求一個串的重複子串)

POJ 2406 Power Strings(KMP求一個串的重複子串)

題意:給一個字串S長度不超過10^6,求最大的n使得S由n個相同的字串a連線而成,如:"ababab"則由n=3個"ab"連線而成,"aaaa"由n=4個"a"連線而成,"abcd"則由n=1個"abcd"連線而成。

定理:假設S的長度為len,則S存在迴圈子串,當且僅當,len可以被len - next[len]整除,最短迴圈子串為S[len - next[len]]

例子證明:
設S=q1q2q3q4q5q6q7q8,並設next[8] = 6,此時str = S[len - next[len]] = q1q2,由字串特徵向量next的定義可知,q1q2q3q4q5q6 = q3q4q5q6

q7q8,即有q1q2=q3q4,q3q4=q5q6,q5q6=q7q8,即q1q2為迴圈子串,且易知為最短迴圈子串。由以上過程可知,若len可以被len - next[len]整除,則S存在迴圈子串,否則不存在。

解法:利用KMP演算法,求字串的特徵向量next,若len可以被len - next[len]整除,則最大迴圈次數n為len/(len - next[len]),否則為1。

#include <stdio.h>
#include <algorithm>
#include <iostream>
#include <string.h>
using namespace std;
#define maxn 2000100
char s[maxn];
int nexts[maxn];
int n,m;
void getnexts()
{
	int j=0;
	int k=-1;
	nexts[0]=-1;
	while(j<m)
	{
		if(k==-1||s[j]==s[k])
		{
			j++;k++;
			nexts[j]=k;
		}
		else k=nexts[k];
	}
}
int main()
{
	while(~scanf("%s",s))
	{
		if(s[0]=='.')break;
		m=strlen(s);
		getnexts();
		int n=m-nexts[m];
		if(n!=m&&m%n==0)
			printf("%d\n",m/n);
		else
			printf("1\n");
	}
    return 0;
}