给定一个数组A,其中包含1,2,...,n的排列.
如果出现的所有数字都是连续数字(可能不是有序的话),则A[i..j]数组的子块A称为有效块A[i..j].
给定一个A= [ 7 3 4 1 2 6 5 8]
有效块的数组[3 4], [1,2], [6,5], [3 4 1 2], [3 4 1 2 6 5], [7 3 4 1 2 6 5], [7 3 4 1 2 6 5 8]
给出O(n log n)算法来计算有效块的数量.
use*_*751 17
这是一个最坏情况的O(n log n)分而治之算法.给定排列的非空子列表,将其划分为左半部分,中间元素和右半部分.递归计算左半部分包含的块数和右半部分包含的块数.现在在O(n)时间内,计算包含中间元素的块数,如下所示.
观察到两个有效块的交集为空或有效块,并且整个置换是有效块.这些事实共同意味着闭包的存在:包含指定(非连续)子序列的唯一最小有效块.将左扩展定义为中间元素的闭包,将右扩展定义为不在右半部分的元素.将右扩展定义为中间元素的闭包和不在左半部分的元素.左扩展(分别是右扩展)相对于子列表关系是完全有序的.它们可以通过简单的工作列表算法按线性时间顺序计算.
现在观察两个重叠有效块的并集本身就是一个有效块.我声称包含中间元素的每个有效块都可以写为由最左边的元素生成的左扩展的并集与由最右边的元素生成的右扩展.要计算此表单的联合,请按递增顺序遍历左侧扩展.保持指向最右边扩展的指针,其最右边的元素不在左扩展的最右边,并且最左边的元素留在左扩展的最左边.由于单调性,这些指针只能移动到更大的扩展,因此总工作是线性的.它们绑定在合格的合作伙伴之上和之下,用于当前的左延伸
C++实现:
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <queue>
#include <stdexcept>
#include <vector>
namespace {
typedef std::vector<int> IntVector;
struct Interval {
int left;
int right;
};
Interval MakeInterval(int left, int right) {
Interval i = {left, right};
return i;
}
typedef std::vector<Interval> IntervalVector;
enum Direction {
kLeft,
kRight,
};
// Finds the valid intervals obtained by starting with [pi[mid],
// pi[mid]] and repeatedly extending in direction dir
//
// O(right_boundary - left_boundary)
void FindExtensions(const IntVector& pi, const IntVector& pi_inv,
int left_boundary, int right_boundary,
Direction dir, IntervalVector* extensions) {
int mid = left_boundary + (right_boundary - left_boundary) / 2;
int left = mid;
int right = mid;
int lower = pi[mid];
int upper = pi[mid];
std::queue<int> worklist;
while (true) {
if (worklist.empty()) {
extensions->push_back(MakeInterval(left, right));
if (dir == kLeft) {
if (left == left_boundary) break;
--left;
worklist.push(left);
} else {
if (right == right_boundary) break;
++right;
worklist.push(right);
}
} else {
int i = worklist.front();
worklist.pop();
if (i < left) {
if (i < left_boundary) break;
for (int j = left - 1; j >= i; --j) worklist.push(j);
left = i;
} else if (right < i) {
if (right_boundary < i) break;
for (int j = right + 1; j <= i; ++j) worklist.push(j);
right = i;
}
int x = pi[i];
if (x < lower) {
for (int y = lower - 1; y > x; --y) worklist.push(pi_inv[y]);
lower = x;
} else if (upper < x) {
for (int y = upper + 1; y < x; ++y) worklist.push(pi_inv[y]);
upper = x;
}
}
}
}
int CountValidRecursive(const IntVector& pi, const IntVector& pi_inv,
int left, int right) {
if (right < left) return 0;
int mid = left + (right - left) / 2;
int count = CountValidRecursive(pi, pi_inv, left, mid - 1) +
CountValidRecursive(pi, pi_inv, mid + 1, right);
IntervalVector left_exts;
FindExtensions(pi, pi_inv, left, right, kLeft, &left_exts);
IntervalVector right_exts;
FindExtensions(pi, pi_inv, left, right, kRight, &right_exts);
typedef IntervalVector::const_iterator IVci;
IVci first_good = right_exts.begin();
IVci first_bad = right_exts.begin();
for (IVci ext = left_exts.begin(); ext != left_exts.end(); ++ext) {
while (first_good != right_exts.end() && first_good->right < ext->right) {
++first_good;
}
if (first_good == right_exts.end()) break;
while (first_bad != right_exts.end() && ext->left <= first_bad->left) {
++first_bad;
}
count += std::distance(first_good, first_bad);
}
return count;
}
// Counts the number of intervals in pi that consist of consecutive
// integers
//
// O(n log n)
int CountValid(const IntVector& pi) {
int n = pi.size();
IntVector pi_inv(n, -1);
// Validate and invert pi
for (int i = 0; i < n; ++i) {
if (pi[i] < 0 || pi[i] >= n || pi_inv[pi[i]] != -1) {
throw std::runtime_error("Invalid permutation of {0, ..., n - 1}");
}
pi_inv[pi[i]] = i;
}
return CountValidRecursive(pi, pi_inv, 0, n - 1);
}
// For testing purposes
int SlowCountValid(const IntVector& pi) {
int count = 0;
int n = pi.size();
for (int left = 0; left < n; ++left) {
for (int right = left; right < n; ++right) {
int lower = pi[left];
int upper = pi[left];
for (int i = left + 1; i <= right; ++i) {
if (pi[i] < lower) {
lower = pi[i];
} else if (pi[i] > upper) {
upper = pi[i];
}
}
if (upper - lower == right - left) ++count;
}
}
return count;
}
} // namespace
int main() {
int n = 10;
IntVector pi(n);
for (int i = 0; i < n; ++i) pi[i] = i;
do {
if (SlowCountValid(pi) != CountValid(pi)) {
fprintf(stderr, "Bad permutation:");
for (IntVector::const_iterator x = pi.begin(); x != pi.end(); ++x) {
fprintf(stderr, " %d", *x);
}
fputc('\n', stderr);
}
} while (std::next_permutation(pi.begin(), pi.end()));
}
Run Code Online (Sandbox Code Playgroud)