AGC006D Median Pyramid Hard

piasticOuO

于 2023-02-11 15:24:09 发布

阅读量126

点赞数

分类专栏：塑料的刷题历程文章标签：算法 c++ Powered by 金山文档

本文链接：https://blog.csdn.net/piasticOuO/article/details/128983188

版权

塑料的刷题历程专栏收录该内容

16 篇文章 1 订阅

订阅专栏

题意

有一个共N层的金字塔，最底层是一个1到2N-1的排列，其上每一层的每个数字皆为它正下方，左下方和右下方共三个数字的中位数。现在给定最底层的排列，求最顶层的数字。

思路

暴力求解是 $\text{[math]}$ 的，无法忍受，下面的二分也不能直接模拟，不然还不如暴力。

能够想到二分是本题解法的关键点之一。将该排列按照一个标准转化为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;
}