求整数k,使得原数组比k大的值都变为k,整个数组的和与目标值target的绝对值差最小。
注意到:k最大只能是max(a[0], a[1]...), 再大的加话,原数组的值也不会变化,最小的是1
我们可以二分来解决这个问题
我们维护的区间(l, r)求出使得数组和比target大的最小k,
但是最优解可能是比target小一点或大一点
就是说
public class Leetcode1300 {
public int findBestValue(int[] arr, int target) {
int l = 1;
int r = 0;
int s = 0;
for (int x : arr) {
s += x;
r = Math.max(r, x);
}
if (s == target) return r;
while (l < r) {
int mid = (l + r) / 2;
int sum = getSum(arr, mid);
if (sum == mid) return mid;
if (sum > target) r = mid;
else l = mid + 1;
}
if (Math.abs(getSum(arr, l - 1) - target) <= Math.abs(getSum(arr, l) - target)) {
return l - 1;
} else {
return l;
}
}
int getSum(int[] arr, int mid) {
int s = 0;
for (int x : arr) {
s += x > mid ? mid : x;
}
return s;
}
}
可能k-1的时候才是最优解,最后check下就好了