C. To Become Max(二分)

传送门

题意

给你一个长度为 $n$的整数数组 $a$。

在一次操作中，你

• 选择一个索引 $i$，使得 $1\le i\le n-1$ 和 $a_i\le a_{i+1}$。
• 用$1$增加$a_i$。

求最多进行 $k$次这一运算后所得到的 $\max (a_1,a_2,\dots a_n)$的最大可能值。

输入

每个测试包含多个测试用例。第一行输入包含一个整数$t$（$1\le t\le 100$）–测试用例数。测试用例说明如下。

每个测试用例的第一行包含两个整数$n$和$k$（$2\le n\le 1000$，$1\le k\le 10^8$）–数组的长度$a$和可执行的最大操作数。

每个测试用例的第二行包含 $n$个整数 $a_1,a_2,\dots ,a_n$（$1\le a_i\le 10^8$）–数组 $a$的元素。

保证所有测试用例的 $n$之和不超过 $1000$。

输出

对每个测试用例输出一个整数，即执行最多 $k$次操作后数组可能的最大值。

思路

二分出数组中每个元素的最大值,然后对这些最大值取个$max$.

思路很清楚,关键是check函数怎么写.

首先我们先假设$x$是数组中某个元素的最大值,那么我们怎么样才能用最小的代价$cnt$才能得到它呢?用贪心的思想来说,就是保证后面的数等于前面的数减一,即$a_i=a_{i-1}-1$,但我们要保持a数组不变,所以我们需要开一个b数组作为a数组的副本,然后依次比较一下$a_i$与$b_i$的大小,如果遇到$a_i\ge b_i$并且$cnt\le k$时就可以返回true,否则返回false.

时间复杂度$n^2$$logn$

可能有点抽象,还是举个例子吧(语文不太好)

比如$n=9,k=11,x=8,pos=4$的时候
数组$a:$ $1,2,3,6,5,4,1,2,4$
数组$b:$ $1,2,3,8,7,6,5,4,3$
代价$:$    $0,0,0,2,2,2,4,2,0$
$cnt=2+2+2+4+2=12$
$cnt>k$,所以返回false

当$n=9,k=11,x=8,pos=2$的时候
数组$a:$ $1,2,3,6,5,4,1,2,4$
数组$b:$ $1,8,7,6,5,4,3,2,1$
代价$:$    $0,6,4,0,0,0,0,0,0$
$cnt=6+4=10$
$cnt\le k$,所以返回true.

完整代码

#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
#define int long long
const int N = 3e5 + 10, mod = 998244353, M = 2 * N;
typedef pair<int, int> PII;
typedef long long ll;
const int INF_MAX = 0x3f3f3f3f3f3f3f3f;
int a[N], b[N], n, k;
bool check(int x, int pos)
{
    b[pos] = x;
    for (int i = pos + 1; i <= n; i++)b[i] = b[i - 1] - 1;
    int cnt = 0;
    for (int i = pos; i <= n; i++)
    {
        if (a[i] >= b[i])
        {
            if (cnt <= k)return true;
            else return false;
        }
        cnt += b[i] - a[i];
    }
    return false;
}
void solve()
{
    cin >> n >> k;
    for (int i = 1; i <= n; i++)cin >> a[i];
    int ans = *max_element(a + 1, a + 1 + n);
    for (int i = 1; i <= n; i++)
    {
        int l = a[i], r = a[i] + k;
        while (l <= r)
        {
            int mid = (l + r) / 2;
            if (check(mid, i))
            {
                ans = max(mid, ans);
                l = mid + 1;
            }
            else r = mid - 1;
        }
    }
    cout << ans << '\n';
}
signed main()
{
    ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    int T = 1;
    cin >> T;
    while (T--)solve();
    return 0;
}