【POJ】2406 Power Strings

http://poj.org/problem?id=2406

题意：给定一个字符串 L,已知这个字符串是由某个字符串 S 重复 R 次而得到的，求 R 的最大值。（长度<=1000000）

#include <cstdio>

#include <cstring>

#include <algorithm>

#include <cmath>

#include <iostream>

using namespace std;

int p[2000000];

char s[2000000];

int main() {

	while(scanf("%s", s+1), s[1]!='.') {

		int j=0, n=strlen(s+1); p[1]=0;

		for(int i=2; i<=n; ++i) {

			while(j && s[j+1]!=s[i]) j=p[j];

			if(s[j+1]==s[i]) ++j;

			p[i]=j;

		}

		if(n%(n-p[n])==0) printf("%d\n", n/(n-p[n]));

		else puts("1");

	}

	return 0;

}

kmp求出next后那么最短循环串的长度为n-next[n]，只需要判断n是否整除它即可。

#include <cstdio>

#include <algorithm>

#include <cstring>

#include <cmath>

using namespace std;



const int N=2000005;

inline void sort(int *x, int *y, int *sa, int n, int m) {

	static int c[N], i;

	for(i=0; i<m; ++i) c[i]=0;

	for(i=0; i<n; ++i) ++c[x[y[i]]];

	for(i=1; i<m; ++i) c[i]+=c[i-1];

	for(i=n-1; i>=0; --i) sa[--c[x[y[i]]]]=y[i];

}

inline void hz(int *a, int *sa, int n, int m) {

	static int t1[N], t2[N], i, j, p, *x, *y, *t;

	x=t1, y=t2;

	for(i=0; i<n; ++i) x[i]=a[i], y[i]=i;

	sort(x, y, sa, n, m);

	for(j=1, p=1; p<n; j<<=1, m=p) {

		p=0;

		for(i=n-j; i<n; ++i) y[p++]=i;

		for(i=0; i<n; ++i) if(sa[i]-j>=0) y[p++]=sa[i]-j;

		sort(x, y, sa, n, m);

		for(t=x, x=y, y=t, p=1, x[sa[0]]=0, i=1; i<n; ++i)

			x[sa[i]]=y[sa[i]]==y[sa[i-1]]&&y[sa[i]+j]==y[sa[i-1]+j]?p-1:p++;

	}

}

inline void geth(int *s, int *sa, int *rank, int *h, int n) {

	static int j, i, k;

	for(i=1; i<=n; ++i) rank[sa[i]]=i;

	for(k=0, i=1; i<=n; h[rank[i++]]=k)

		for(k?--k:0, j=sa[rank[i]-1]; s[i+k]==s[j+k]; ++k);

}



int a[N], sa[N], h[N], rank[N], n;

char s[N];

inline int work() {

	static int len[N], pos;

	pos=rank[1];

	for(int i=pos, mn=h[i]; i>1; --i) mn=min(h[i], mn), len[i-1]=mn;

	for(int i=pos+1, mn=h[i]; i<=n; ++i) mn=min(h[i], mn), len[i]=mn;

	int sq=sqrt(n+0.5);

	for(int k=1; k<=sq; ++k) if(n%k==0 && len[rank[k+1]]==n-k) return n/k;

	return 1;

}

int main() {

	while(scanf("%s", s+1), s[1]!='.') {

		n=strlen(s+1);

		for(int i=1; i<=n; ++i) a[i]=s[i];

		a[0]=0;

		hz(a, sa, n+1, 255);

		geth(a, sa, rank, h, n);

		printf("%d\n", work());

	}

	return 0;

}

　　

---

 

经典题...可是我tle了......因为本题听说是用kmp.............QAQ

sa的做法就是，求出height后，我们只匹配suffix(1)和suffix(k+1)的最长公共前缀是否为n-k即可，k是枚举的长度...至于为什么，请自己想...