这里是刚学c语言没多久的菜鸟一只
这是从我同学那借鉴(抄)来的代码 进行了注释 分享给跟我这个阶段一样的人
重要的思想就是二分查找,而这个代码个人觉得妙在 “while(!check(r + 1,x,arr,n)) {r += 1;}”这个代码 能在最后从比答案稍小的位置一一加上去精确查找到最终答案
//
// Created by Mrlaolu on 2023/9/22.
//
#include <bits/stdc++.h>
bool check(int h,long long x,long long arr[],long long n) //检查空位是否超过总水量
{
long long sum = 0;
for (long long i = 0; i < n; ++i)
{
if (h - arr[i] > 0)sum += h - arr[i];
}
return sum > x;
}
int main()
{
long long t;
scanf("%lld",&t);
while(t--)
{
long long n = 0,x = 0;
scanf("%lld%lld",&n,&x);
long long arr[201000],take = 0;
long long l = x / n; // 给l设个初始值,h不会小于x/n(完全无珊瑚的情况还舍去小数点)
for(long long i = 0;i < n; ++i)
{
scanf("%lld",&arr[i]);
take += arr[i];
}
long long r = (take + x) / n,mid = 0; //给r设个初始值,h不会大于(x+take/n)(珊瑚完全平铺的情况还舍去小数点)
while(l <= r) //二分查找
{
mid = (l + r) / 2;
if(check(mid,x,arr,n)){r = mid - 1;} //check函数用于检查h = mid 时,空位有没有超出水位
else {l = mid + 1;} //最终出循环的条件是 l > r ,目的是下面那条代码
}
while(!check(r + 1,x,arr,n)) {r += 1;} //让r从小往大依次找 , 使不漏掉出现的最大的坐标
printf("%lld\n",r);
}
return 0;
}