POJ2406 Power Strings 【KMP 或 后缀数组】

电源串

时间限制： 3000MS		内存限制： 65536K
提交总数： 53037		接受： 22108

描述

给定两个字符串a和b，我们定义a * b是它们的连接。例如，如果a =“abc”和b =“def”，那么a * b =“abcdef”。如果我们将连接看作是乘法，则用正常的方式定义非负整数的指数：a ^ 0 =“”（空字符串）和a ^（n + 1）= a *（a ^ n）。

输入

每个测试用例都是一行代表s的输入，一串可打印的字符。s的长度至少为1，不会超过100万字符。在最后一个测试用例之后包含句点的行。

产量

对于每一个s，你应该打印出最大的n，这样对于某个字符串a，s = a ^ n。

示例输入

A B C D
AAAA
ABABAB
。

示例输出

1
4
3

暗示

这个问题有巨大的投入，使用scanf而不是cin来避免超时。

题解

这道题正解应该是KMP，我之前的KMP复习的博客里有解释，这里就不赘述了

主要是根据如果最后一位的最长等于的后缀的前缀长与总长的差刚好是len的因子，说明len就是最大循环节

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define fo(i,x,y) for (int i = (x); i <= (y); i++)
#define Redge(u) for (int k = head[u]; k != -1; k = edge[k].next)
using namespace std;
const int maxn = 1000005,maxm = 100005,INF = 1000000000;

char P[maxn];
int n,f[maxn];

void getf(){
	int j = f[0] = -1,i = 0;
	while (i < n){
		while (j != -1 && P[j] != P[i]) j = f[j];
		f[++i] = ++j;
	}
}

int main()
{
	while (true){
		fgets(P,maxn,stdin);
		n = strlen(P) - 1;
		if (n == 1 && P[0] == '.') break;
		getf();
		if (n % (n - f[n]) == 0) printf("%d
",n/(n - f[n]));
		else printf("1
");
	}
	return 0;
}

为什么突发奇想又写一次这道题呢？

因为今天在练后缀数组时看到有博主用后缀数组写了，不过倍增的后缀数组会T  QAQ【我只会倍增求法】

但也拍上来，主要是理解了lcp的求法

代码【非正解】：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define Redge(u) for (int k = head[u]; k != -1; k = edge[k].next)
using namespace std;
const int maxn = 1000005,maxm = 10005,INF = 1000000000;
inline int RD(){
	int out = 0,flag = 1; char c = getchar();
	while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}
	while (c >= 48 && c <= 57) {out = (out << 1) + (out << 3) + c - '0'; c = getchar();}
	return out * flag;
}
char s[maxn];
int n,sa[maxn],rank[maxn],height[maxn],t1[maxn],t2[maxn],m = 300,c[maxn];
int lcp[maxn];
void SA(){
	int *x = t1,*y = t2;
	for (int i = 0; i <= m; i++) c[i] = 0;
	for (int i = 1; i <= n; i++) c[x[i] = s[i]]++;
	for (int i = 1; i <= m; i++) c[i] += c[i - 1];
	for (int i = n; i >= 1; i--) sa[c[x[i]]--] = i;
	for (int k = 1; k <= n; k <<= 1){
		int p = 0;
		for (int i = n - k + 1; i <= n; i++) y[++p] = i;
		for (int i = 1; i <= n; i++) if (sa[i] - k > 0) y[++p] = sa[i] - k;
		for (int i = 0; i <= m; i++) c[i] = 0;
		for (int i = 1; i <= n; i++) c[x[y[i]]]++;
		for (int i = 1; i <= m; i++) c[i] += c[i - 1];
		for (int i = n; i >= 1; i--) sa[c[x[y[i]]]--] = y[i];
		swap(x,y);
		p = 1; x[sa[1]] = 1;
		for (int i = 2; i <= n; i++)
			x[sa[i]] = (y[sa[i]] == y[sa[i - 1]] && y[sa[i] + k] == y[sa[i - 1] + k]) ? p : ++p;
		if (p >= n) break;
		m = p;
	}
	for (int i = 1; i <= n; i++) rank[sa[i]] = i;
	int k = 0;
	for (int i = 1; i <= n; i++){
		if (k) k--;
		int j = sa[rank[i] - 1];
		while (s[i + k] == s[j + k]) k++;
		height[rank[i]] = k;
	}
}
void LCP(){
	int k = rank[1]; lcp[k] = n;
	for (int i = k - 1; i >= 1; i--) lcp[i] = min(lcp[i + 1],height[i + 1]);
	for (int i = k + 1; i <= n; i++) lcp[i] = min(lcp[i - 1],height[i]);
}
bool check(int len){return lcp[rank[len + 1]] == n - len;}
int main(){
	while (true){
		m = 300;
		scanf("%s",s + 1); if(s[1] == '.') break;
		n = strlen(s + 1);
		SA();
		LCP();
		int E = (int)sqrt(n),ans = 0;
		for (int i = 1; i <= E; i++){
			if (n % i) continue;
			if (check(i)) ans = max(ans,n / i);
			if (check(n / i)) ans = max(ans,i);
		}
		printf("%d
",ans);
	}
	return 0;
}