本文主要是介绍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
解题思路
完全没有思路。。。看官方题解做的
version1: 枚举value。首先先用一个dict保存arr的信息,以arr中每个数字作为key,value是tuple,保存所有比key小的数字之和,以及比key大的数字个数。然后从0开始寻找value,每次的和是tuple[0] + tuple[1] * value。注意保存prev_sum和cur_sum,如果prev_sum <= target <= cur_sum,则找到value了
version2: 用二分法找value。思路和二分查找没有太大区别,需要注意的地方有2点:
- 比较的数字不再是given array,而是array的和,所以写一个转换函数比较方便
- 如果没找着,那么就要从left和right里选一个更符合要求的
成绩
version1
时间>5.14%
空间>25.00%
version2
时间>72.57%
空间>25.00%
代码
version1:
class Solution:def findBestValue(self, arr: List[int], target: int) -> int:arr.append(0)arr.sort()arr_smaller_dict = {} # store tuple: (sum of numbers in arr smaller than key, nums of numbers in arr larger than key)for index, each_num in enumerate(arr):arr_smaller_dict[each_num] = (sum(arr[:index + 1]), len(arr) - 1 - index)cur_arr_sum = 0arr_index = 0for value in range(1, max(arr) + 1):prev_arr_sum = cur_arr_sum while arr[arr_index] < value:arr_index += 1smaller_sum, larger_num = arr_smaller_dict[arr[arr_index - 1]][0], arr_smaller_dict[arr[arr_index - 1]][1]cur_arr_sum = smaller_sum + value * larger_numif prev_arr_sum <= target <= cur_arr_sum:breakif abs(prev_arr_sum - target) <= abs(cur_arr_sum - target):value -= 1return value
version2:
class Solution:def findBestValue(self, arr: List[int], target: int) -> int:arr.sort()def arr_to_sum(value: int) -> int:for index, each_num in enumerate(arr):if each_num > value:breakreturn sum(arr[:index]) + value * (len(arr) - index)left, right = 0, arr[-1]while left <= right:middle = (left + right) // 2middle_num = arr_to_sum(middle)if middle_num == target:return middleif middle_num > target:right = middle - 1else:left = middle + 1if abs(arr_to_sum(left) - target) < abs(arr_to_sum(right) - target):value = leftelif abs(arr_to_sum(left) - target) > abs(arr_to_sum(right) - target):value = rightelse:value = min(left, right)return value
这篇关于leetcode-1300. 转变数组后最接近目标值的数组和的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!