题意
有一个共N层的金字塔,最底层是一个1到2N-1的排列,其上每一层的每个数字皆为它正下方,左下方和右下方共三个数字的中位数。现在给定最底层的排列,求最顶层的数字。
思路
暴力求解是的,无法忍受,下面的二分也不能直接模拟,不然还不如暴力。
能够想到二分是本题解法的关键点之一。将该排列按照一个标准转化为01串,然后二分这个标准,此时必定存在一个数,标准大于等于这个数时顶上的数为0,标准小于这个数时为1。1111100000也满足单调,因此可以用二分来求解。
假设我们将这个排列转化为一个01串:11100010101011100......很容易能想到:如果这三个数的和大于等于2(也就是两个1),那么上面那个数就是1,否则就是0。
好像没什么思路了,那就手玩一下吧。注意观察 "11"的情况:
1 | ||||||
1 | 1 | 0 | ||||
1 | 1 | 1 | 0 | 0 | ||
1 | 0 | 1 | 1 | 0 | 0 | 0 |
吃饭的时候突然想到:如果场上存在11,那么无论11左右是1或0,11上面一定是11(除非边界)这个规律对00也适用(可在上表看到00的情况),那么就存在以下情况:
小小的分类讨论?
情况1:只有11,没有00(或者只有00,没有11).比如:11010101010101010101...
首先我们要明白1010101010101......的时候究竟会产出什么数字,所以先跳转到情况2——
情况2(实现时可能需要特判):没有11,也没有00,如:10101010101...... 和 010101010101.....
101 -> 1
10101 -> 101
1010101010101... -> 1010101....
所以我们可以总结得到规律了:情况2的时候每次这一大串都会缩一下,但是整体结构是不变的,所以若最底层是101010101... 最顶层下面就是101 最顶层就是1,所以这个情况下答案就是第一个数啦!
好,现在跳转到情况1:
首先11010101010101.... = 11 + 010101010101... 此时最后一个数字是0,考虑边缘(11010)容易发现这里的结果是110,即110101010101... -> 1101010101... 因此最后会变成110 答案是1
00 + 1010101... 情况也是同理。
情况3:有11,也有00。
我们再次看回到上面的表格,可以发现尽管0更多,但是1却处于黄金地带——换句话说,11离中间更近,所以也就会占到更好的位置。考虑1100...(不局限于最底层!情况3一定会存在这种情形!)在边界时11只能占有一个1,而00有2个0,因此00更占便宜,因此得到规律:谁更接近中间最顶层就是谁!
实现
用一次左右开弓式的遍历来获取离中间更近的11(00)!
int final = d[1]; //d[]是已经处理好的01串
for (int i = 1; i < n; ++i) {
if (d[i] == d[i + 1]) {
final = d[i];
}
if (d[2 * n - i] == d[2 * n - i - 1]) {
final = d[2 * n - i];
}
}
而二分和处理在实现上相信看到这里的你已经十分熟悉了,这里不做赘述。
代码
#include <iostream>
#define MAXN 100005
using namespace std;
int a[MAXN << 1], d[MAXN << 1];
int ans, n;
bool judge(int x) {
for (int i = 1; i < 2 * n; ++i) {
d[i] = (a[i] >= x);
}
int final = d[1];
for (int i = 1; i < n; ++i) {
if (d[i] == d[i + 1]) {
final = d[i];
}
if (d[2 * n - i] == d[2 * n - i - 1]) {
final = d[2 * n - i];
}
}
return final;
}
int main() {
cin >> n;
for (int i = 1; i < 2 * n; ++i) {
cin >> a[i];
}
int l = 1, r = 2 * n - 1;
while (l <= r) {
int mid = l + ((r - l) >> 1);
if (judge(mid)) {
ans = mid;
l = mid + 1;
} else {
r = mid - 1;
}
}
cout << ans;
return 0;
}