Leetcode1300—— 转变数组后最接近目标值的数组和

考点:双重二分

函数:

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& arr, int target) {
            int len=arr.size();
            vectorprefix(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;
            
    }
};

 

你可能感兴趣的:(Leetcode)