考点:双重二分
函数:
max_element(vec.begin(),vec.end())返回指向最大元素的迭代器
lower_bound(vec.begin(),vec.end(),val)返回指向排好序数组的大于等于val的第一个元素的迭代器
lower_bound(vec.begin(),vec.end(),val)返回指向排好序数组的大于val的第一个元素的迭代器
class Solution {
public:
int findBestValue(vector<int>& arr, int target) {
int len=arr.size();
vector<int>prefix(len+1);
int ans=-1;
sort(arr.begin(),arr.end());
int l=0;
int r=*max_element(arr.begin(),arr.end());
for(int i=1;i<=len;i++)
{
prefix[i]=prefix[i-1]+arr[i-1];
}
while(l<=r)
{
int mid=(l+r)/2;
auto iter=lower_bound(arr.begin(),arr.end(),mid);
int num=prefix[iter-arr.begin()]+(arr.end()-iter)*mid;
if(num>target)
{
r=mid-1;
}
else
{
ans=mid;
l=mid+1;
}
}
auto iter=lower_bound(arr.begin(),arr.end(),ans);
int choose_small=prefix[iter-arr.begin()]+(arr.end()-iter)*ans;
iter=lower_bound(arr.begin(),arr.end(),ans+1);
int choose_big=prefix[iter-arr.begin()]+(arr.end()-iter)*(ans+1);
return abs(choose_small-target)<=abs(choose_big-target)?ans:ans+1;
}
};