[1300. 转变数组后最接近目标值的数组和]
难度 中等
给你一个整数数组 arr
和一个目标值 target
,请你返回一个整数 value
,使得将数组中所有大于 value
的值变成 value
后,数组的和最接近 target
(最接近表示两者之差的绝对值最小)。
如果有多种使得和最接近 target
的方案,请你返回这些整数中的最小值。
请注意,答案不一定是 arr
中的数字。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-mutated-array-closest-to-target
示例 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
解法:二分查找
value 值的上下界分析:
- value 的下界为 0。因为当 value = 0 时,数组的和为 0。由于 target 是正整数,因此当 value 继续减小时,数组的和也会随之减小,且变为负数,并不会比 value = 0 时更接近 target。
- value 的上界为数组 arr 中的最大值。这是因为当 value >= arr 时,数组中所有的元素都不变,因为它们均不大于 value。
随着 value 的增大,数组的和是严格单调递增的。这里「严格」的意思是,不存在两个不同的 value 值,它们对应的数组的和相等。因此我们可以采用 二分查找 找到一个 valueSmall,使得数组的和最接近 target 且小于 target。同样我们可以找到一个 valueBig,使得数组的和最接近 target 且大于 target(valueBig = valueSmall + 1)。最后比较采用 valueSmall 和 valueBig 所获得的数组元素之和与 target 的接近程度,并返回最接近的那个 value
class Solution {
public int findBestValue(int[] arr, int target) {
//如果 target = 1,则 value 的值为 0,
//因为 1 <= arr[i], value 增大后 改变后的数组和 和 target 之差的绝对值增大
if(target == 1) {
return 0;
}
int sum = 0; //数组所有元素和
int maxValue = Integer.MIN_VALUE; //数组中的最大元素值
for (int x : arr) {
sum += x;
if(maxValue < x) {
maxValue = x;
}
}
//如果 target > 数组中所有元素之和, 则 value 的值为数组中的最大元素
if(sum < target) {
return maxValue;
}
int ans = -1;
int left = 1, right = maxValue;
while(left <= right) {
int mid = left + (right - left)/2;
//获取改变后的数组元素和
int currentSum = getCurrentSum(arr, mid);
if (currentSum <= target) {
ans = mid;
left = mid + 1;
}
else {
right = mid - 1;
}
}
int valueSmall = getCurrentSum(arr, ans);
int valueBig = getCurrentSum(arr, ans + 1);
if(target - valueSmall > valueBig - target) {
return ans + 1;
}
return ans;
}
//获取改变后的数组元素和
public int getCurrentSum(int[] arr, int value) {
int currentSum = 0;
for (int x : arr) {
currentSum += Math.min(x, value);
}
return currentSum;
}
}