POJ 3581 Sequence[后缀数组]：一串三分求最小

题目大意

题目链接
给N个数(N不超过200000), 保证第一个数比后面的都大, 要求你把数列分成连续的三段, 分别逆序再拼起来, 要求新的串字典序最小.

思路分析

一个直接的思路是想把最小的一个数作为第一段的末尾, 这样是不是最优的呢? 是的, 因为第一个数比后面的数都大, 于是你把所有的最小的数作为第一段的末尾逆序时, 最大的数之前的部分的字典序肯定不一样. 于是就先把整个序列逆序, 求一个后缀数组, 按i从小到大考虑sa[i], 因为第一个串的长度最多是N - 2, 所以sa[i]必须大于2, 找到最小的i即可.

于是问题转化为, 一个数字串, 分成两段并分别逆序, 使得字典序最小. 假设序列为s, 长度为m, 第一串的末尾为k, 则原来的串s1s2…sk…sk+1sk+2…sm, 在操作后变成了sksk-1…s1smsm-1…sk+1, 注意到这个串是smsm-1sm-2…s1smsm-1sm-2…s1的子串, 于是做法就明了了: 把s逆序并在后面复制一遍, 做一次后缀数组, 找到最小的i, 使得i不超过m, 且要求它大于1(要保证是两个串). 于是时间复杂度O(NlogN), 其实就是后缀数组的复杂度, 我写的倍增啦~

#include<iostream>
#include<string>
#include<string.h>
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;

#define MAX 200005

int sa[MAX * 2], Rank[MAX * 2], tmp[MAX * 2];
int n, k, v[MAX * 2], t[MAX];

bool cmp(int i, int j) {
	if (Rank[i] != Rank[j])return Rank[i] < Rank[j];
	else {
		int ri = i + k <= n ? Rank[i + k] : -1;
		int rj = j + k <= n ? Rank[j + k] : -1;
		return ri < rj;
	}
}

void build(int n) {
	for (int i = 0; i <= n; i++) {
		sa[i] = i;
		Rank[i] = i < n ? v[i] : -1;
	}

	for (k = 1; k <= n; k *= 2) {
		sort(sa, sa + n + 1, cmp);
		tmp[sa[0]] = 0;
		for (int i = 1; i <= n; i++) {
			tmp[sa[i]] = tmp[sa[i - 1]] + (cmp(sa[i - 1], sa[i]) ? 1 : 0);
		}
		for (int i = 0; i <= n; i++)Rank[i] = tmp[i];
	}
}
int main() {
	scanf("%d", &n);
	for (int i = 0; i < n; i++)scanf("%d", &t[i]);
	reverse_copy(t, t + n, v);
	build(n);
	int p1; //确定第一段的分割位置
	for (int i = 0; i <= n; i++) {
		p1 = n - sa[i];//sa[i]是rank为i的后缀数组的开始位置 但是这是reverse过的
		if (p1 >= 1 && n - p1 >= 2)break;//第一段不为空，至少剩余两个元素
	}

	// 将pi之后的字符串反转并重复2次，再计算其后缀数组
	int m = n - p1;
	reverse_copy(t + p1, t + n, v);
	reverse_copy(t + p1, t + n, v + m);
	build(2 * m);//注意这里长度要变化
	int p2;
	// 确定后两段的分割位置，注意i小于等于  因为rank包含了空串
	for (int i = 0; i <= 2 * m; i++) {
		//位置的确定 m-sa[i]表示他在不反转子串中的位置 p1+m-sa[i]即在原完整子串的位置
		p2 = p1 + m - sa[i];
		if (p2 - p1 >= 1 && n - p2 >= 1)break;
	}
	reverse(t, t + p1);
	reverse(t + p1, t + p2);
	reverse(t + p2, t + n);
	for (int i = 0; i < n; i++)printf("%d\n", t[i]);
}