题目详情
给你一个整数数组 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) {
/*对arr进行排序*/
sort(arr.begin(), arr.end());
/*预处理 数组arr的前缀和*/
int n = arr.size();
vector<int> save(n + 1);
for (int i = 1; i <= n; i++)
{
save[i] = save[i - 1] + arr[i - 1];
}
/*二分法找出value_lower*/
int l = 0;
int r = *max_element(arr.begin(), arr.end());
int value_lower = 0;
while (l <= r) {
int mid = (l + r) / 2;
int index = my_lower_bound(arr, mid); //index存储的是第一个不小于mid的arr元素的下标
int cur = save[index] + (n - index) * mid;
if (cur <= target) {
value_lower = mid;
l = mid + 1;
}
else {
r = mid - 1;
}
}
//value_lower为value的下界
//value_upper为value的上界
/*比较value_lower 和 value_upper 所对应的数组和*/
int index_lower = my_lower_bound(arr, value_lower);
int choose_lower = save[index_lower] + (n - index_lower) * value_lower;
int value_upper = value_lower + 1;
int index_upper = my_lower_bound(arr, value_upper);
int choose_upper = save[index_upper] + (n - index_upper) * value_upper;
return abs(choose_lower - target) <= abs(choose_upper - target) ? value_lower : value_upper;
}
/*二分法找出第一个不小于val的arr元素的下标*/
int my_lower_bound(vector<int>& arr, int val) {
int l = 0;
int r = arr.size() - 1;
int index = 0;
while (l <= r) {
int mid = (l + r) / 2;
if (arr[mid] >= val) {
index = mid;
r = mid - 1;
}
else {
l = mid + 1;
}
}
return index;
}
};