1300. 转变数组后最接近目标值的数组和. 二分

给你一个整数数组 arr 和一个目标值 target ,请你返回一个整数 value ,使得将数组中所有大于 value 的值变成 value 后,数组的和最接近  target (最接近表示两者之差的绝对值最小)。

如果有多种使得和最接近 target 的方案,请你返回这些整数中的最小值。

请注意,答案不一定是 arr 中的数字。

示例 1:

输入:arr = [4,9,3], target = 10
输出:3
解释:当选择 value 为 3 时,数组会变成 [3, 3, 3],和为 9 ,这是最接近 target 的方案。
示例 2:

输入:arr = [2,3,5], target = 10
输出:5
示例 3:

输入:arr = [60864,25176,27249,21296,20204], target = 56803
输出:11361

提示:

1 <= arr.length <= 10^4
1 <= arr[i], target <= 10^5

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-mutated-array-closest-to-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    int getsum(vector<int>& arr, vector<int>& presum, int val){
        auto iter = lower_bound(arr.begin(), arr.end(), val);
        int ret = presum[iter - arr.begin()] + val * (arr.end() - iter);
        return ret;
    }

    int findBestValue(vector<int>& arr, int target) {
        //排序
        int n = arr.size();
        sort(arr.begin(), arr.end());
        //前缀和
        vector <int> presum(n + 1);
        for (int i = 1;i <= n; i++){
            presum[i] = presum[i - 1] + arr[i - 1];
        }
        //二分
        int l = 0,r = arr[n - 1];
        while (l <= r){
            int mid = (l + r) >> 1;
            if (getsum(arr, presum, mid) < target){
                l = mid + 1;
            }
            else{
                r = mid - 1;
            }
        }
        
        int x = abs(getsum(arr, presum, l) - target);
        int y = abs(getsum(arr, presum, l - 1) - target);
        if (x < y){
            return l;
        }
        else{
            return l - 1;
        }
    }
};
posted @ 2020-06-14 22:15  _西瓜不甜  阅读(123)  评论(0)    收藏  举报