题目 【传送门】
你需要修一个水池,水池的底部高低不一,第 i 格高 a [i] 。
如果修了一个高 h 的水池,会先在两侧修高度为 h 的墙壁防止水溢出,然后对每一格注水:
- 对于第 i 格,需要向其注水 h−ai 个单位的水,如果 a[i] >h,则不用注水。
- 水可以有剩余,但不能少,少了就无法修建高度为 h 的水池。
你一共有 x 个单位的水,问 h 最大能是多少。
分析
我们发现,当h越小时,所需的水越少;当h越大时,所需的水越多; h大小与所需水量的关系满足单调性,故可以使用二分
每次二分时,设mid代入,进行check,满足则表示h可以取到更大,继续二分
Code
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,m,t,l,r,x,a[200005],s[200005];
bool check(ll h)
{
ll sum=0;
for(int i=1;i<=n;i++) sum+=max(0ll,h-a[i]);
return (sum<=m);
}
int main()
{
scanf("%ld",&t);
while(t--)
{
scanf("%lld%lld",&n,&m);
memset(s,0,sizeof s); memset(a,0,sizeof a);
for(int i=1;i<=n;i++) scanf("%lld",a+i);
for(int i=1;i<=n;i++) s[i]=s[i-1]+a[i];
l=0,r=2e9,x=0;
while(l<=r)
{
int mid=l+(r-l)/2;
if(check(mid)) x=mid,l=mid+1;
else r=mid-1;
}
cout<<x<<"\n";
}
return 0;
}