给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-mutated-array-closest-to-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
根据题意可以猜到答案必定落在区间【0, max(arr)】,
而且对于每一个猜测的答案,都可以遍历整个数组得到新的数组之和 tmp,并根据 tmp 和target的关系来调整区间大小。
所以此题是用二分试探解题。
时间复杂度:O(NlogN)
空间复杂度:O(1)
class Solution(object):
def findBestValue(self, arr, target):
"""
:type arr: List[int]
:type target: int
:rtype: int
"""
left, right = 0, max(arr)
sub, ans = float("inf"), float("inf")
while left <= right:
mid = (left + right) // 2 # mid 为本次猜测的答案
tmp = 0 # tmp 是本次猜测对应新数组之和
for num in arr:
if num > mid:
tmp += mid
else:
tmp += num
cur_sub = abs(tmp - target) #当前差的最小值
if cur_sub < sub: # 如果有更小的答案
sub = cur_sub
ans = mid
elif cur_sub == sub:
ans = min(ans, mid)
if tmp > target: # 调整区间大小
right = mid - 1
elif tmp < target:
left = mid + 1
elif tmp == target:
return mid
return ans