Codeforces Round #898 (Div. 4) 1873E - Building an Aquarium 二分查找 c语言写法

这里是刚学c语言没多久的菜鸟一只

这是从我同学那借鉴(抄)来的代码 进行了注释 分享给跟我这个阶段一样的人 

重要的思想就是二分查找,而这个代码个人觉得妙在 “while(!check(r + 1,x,arr,n)) {r += 1;}”这个代码 能在最后从比答案稍小的位置一一加上去精确查找到最终答案

//
// Created by Mrlaolu on 2023/9/22.
//

#include 
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;
}

你可能感兴趣的:(c++,算法)