1300. 转变数组后最接近目标值的数组和(C++)---双重二分查找解题

题目详情
 

给你一个整数数组 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;
	}
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章