【每日一题】补档 CF1623C. Balanced Stone Heaps | 二分答案 | 中等

题目内容

原题链接

给定一个长度为 $n$ 的数组 $a$ ，下标从 $1$ 开始。

从下标 $index=3$ 开始从前往后操作，可以选择将 $a[index]$ 减去 $3k(3k\le a[index])$
然后将 $a[index-1]$ 加上 $k$ ，$a[index-2]$ 加上 $2k$ ，也可以不操作。

问操作后的数组 $a$ ，其最小值最大为多少。

数据范围

• $3\le n\le 2\cdot 10^5$
• $1\le a_i\le 10^9$

题解

二分答案为 $mid$。

$check$ 函数如何考虑？

需要将每个数都上升至 $mid$ ，而第 $i$ 个数的上升只能由第 $i+1$ 个和第 $i+2$ 个数操作而来。

但是这么考虑有一个问题在于，如果第 $i$ 个数的上升了 $x$ ，那么多少来自于第 $i+1$ 个数，多少来自于第 $i+2$ 个数？

我们可以枚举有 $a$ 来自第 $i+1$ 个数，$x-a$ 来自第 $i+2$ 个数，但是值域太大了，这样必然超时。

如果反过来考虑呢？

对于第 $n$ 个数，如果 $a[n]>mid$ ，那么多余的部分可以给到 $a[n-1]$ 和 $a[n]$ ，假设 $a[n]$ 降低 $3k$ ，那么 $a[n-1]$ 可以增加 $k$ ，$a[n-2]$ 增加了 $2k$ 。

但是你需要注意的是，由后面的数增加而来的部分，是不能拿来下降再给其之前的两个数上升的。

所以我们可以将由之后的数下降而来的部分，单独开一个 $extra$ 部分。

但是下降的部分只能是 $a$ 本身的部分。

这样我们反着考虑，最后只有第 $1$ 个数和第 $2$ 个数是不能用来给前面的数上升的，判断所有的最终 $a_i+extr{a}_i$ 是否都大于等于 $mid$ 即可。

时间复杂度：$O(n\log (2\times 10^{14}))$ ，其中 ${\log }_2(2\times 10^{14})<48$

代码

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

void solve() {
    int n;
    cin >> n;

    vector<ll> a(n);
    for (int i = 0; i < n; ++i) cin >> a[i];

    vector<ll> b(n);
    vector<ll> extra(n);
    auto check = [&](ll mid) {
        for (int i = 0; i < n; ++i) b[i] = a[i], extra[i] = 0;

        for (int i = n - 1; i >= 2; --i) {
            if (b[i] + extra[i] < mid) return false;
            ll ch = min(b[i], b[i] + extra[i] - mid);
            ch -= ch % 3;
            extra[i - 1] += ch / 3;
            extra[i - 2] += ch / 3 * 2;
            b[i] -= ch;
        }

        return b[0] + extra[0] >= mid && b[1] + extra[1] >= mid;
    };

    ll l = 1, r = 2e14;
    while (l < r) {
        ll mid = (l + r + 1) >> 1;
        if (check(mid)) l = mid;
        else r = mid - 1;
    }

    cout << l << "\n";
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int T = 1;
    cin >> T;

    while (T--) {
        solve();
    }

    return 0;
}