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

leetcode 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

我的代码

class Solution {
public:
    int findBestValue(vector<int>& arr, int target) {
        sort(arr.begin(), arr.end());
        int len = arr.size();
        vector<long> sums(len);
        sums[0] = arr[0];
        for (int i = 1; i < len; ++i) {
            sums[i] = sums[i - 1] + arr[i];
        }
        if (target <= arr[0] * len) {
            double remain = (target % len * 1.0) / len;
            return target / len + (remain <= 0.5 ? 0 : 1);
        }
        if (sums[len - 1] <= target) {
            return arr[len - 1];
        }
        int i;
        for (i = 0; i < len; ++i) {
            if (sums[i] + arr[i] * (len - i - 1) > target) {
                break;
            }
        }
        int need = target - (int)sums[i - 1];
        double remain = (need % (len - i) * 1.0) / (len - i);
        return need / (len - i) + (remain <= 0.5 ? 0 : 1);
    }
};

我的成绩

执行结果：通过
执行用时 : 76 ms, 在所有 C++ 提交中击败了15.69%的用户
内存消耗 : 13.1 MB, 在所有 C++ 提交中击败了11.11%的用户

一些想法

本道题我是按数学的方法做的，首先将数组排序，然后对边界进行处理，再遍历排好序的数组，找到大于target的最小值，再通过数学方法计算出返回值即可。

执行用时为 8 ms 的范例

class Solution {
public:
    int sumFunc(vector<int>& arr, int threshold) {
        int sum = 0;
        for (int i = 0; i < arr.size(); i++) {
            if (arr[i] <= threshold) {
                sum += arr[i];
            } else {
                sum += threshold;
            }
        }
        return sum;
    }
    int findBestValue(vector<int>& arr, int target) {
        int left = 1;
        int right = *max_element(arr.begin(), arr.end());  // 二分法上下限分别是0和max(arr)
        while (left < right) {
            int mid = left + (right - left) / 2;
            std::cout<<"mid = "<<mid<<std::endl;
            int sum = sumFunc(arr, mid);
            if (sum < target) {
                left = mid + 1;
            } else {
                right = mid;
            } 
            std::cout<<"left = "<<left<<std::endl;
            std::cout<<"right = "<<right<<std::endl;
        }
        std::cout<<"left = "<<left<<std::endl;
        int sum1 = sumFunc(arr, left - 1);
        int sum2 = sumFunc(arr, left);
        if (sum2 - target >= target - sum1) {
            return left - 1;
        } else {
            return left;
        }
    }
};

思考

范例是用二分法不断逼近target。