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

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

思路

想到了两分查找，先将数组进行排序，找出二分查找的上界（数组元素最大值）和下界（0），然后进行二分查找，mid = (l + r) / 2，并将数组内大于等于mid的数全部设置为mid，得到数组的和cur，由于数组和必定单调递增（随着mid的增大），因而可以通过二分法找到最接近target又不大于target的value，而value+1必定使数组和最接近target并且大于target，再比较value的数组和与value+1的数组和，谁和target相差最小即可。

代码

class Solution {
public:
    int findBestValue(vector<int>& arr, int target) {
        sort(arr.begin(), arr.end());
        int n = arr.size();
        vector<int> prefix(n+1, 0);
        //求数组前缀和
        for(int i = 1; i <= n; i++) {
            prefix[i] = prefix[i - 1] + arr[i - 1];
        }
        //找到二分查找的上下界
        int l = 0, r = *max_element(arr.begin(), arr.end()), ans = -1;
        while(l <= r) {
            int mid = (l + r) / 2;
            //找到数组中第一个大于或等于mid的元素
            auto iter = lower_bound(arr.begin(), arr.end(), mid);
            //将数组大于等于mid的元素全部设置为mid，小于mid的元素不变，求和
            int cur = prefix[iter - arr.begin()] + (arr.end() - iter) * mid;
            //求的和小于target即将左边界右移
            //否则右边界左移
            if(cur <= target) {
                ans = mid;
                l = mid + 1;
            }
            else {
                r = mid - 1;
            }
        }
        //求ans对应的数组和以及ans+1对应的数组和
        int small = check(arr, ans);
        int big = check(arr, ans + 1);
        return abs(small - target) <= abs(big - target)? ans : ans + 1;
    }
    int check(vector<int>& arr, int x) {
        int ans = 0;
        for(const int& num : arr) {
            ans += num >= x? x : num;
        }
        return ans;
    }
};