Leetcode 1300.转变数组后最接近目标值的数组和
1 题目描述(Leetcode题目链接)
给你一个整数数组 arr 和一个目标值 target ,请你返回一个整数 value ,使得将数组中所有大于 value 的值变成 value 后,数组的和最接近 target (最接近表示两者之差的绝对值最小)。
如果有多种使得和最接近 target 的方案,请你返回这些整数中的最小值。
请注意,答案不一定是 arr 中的数字。
输入:arr = [4,9,3], target = 10
输出:3
解释:当选择 value 为 3 时,数组会变成 [3, 3, 3],和为 9 ,这是最接近 target 的方案。
输入:arr = [2,3,5], target = 10
输出:5
输入:arr = [60864,25176,27249,21296,20204], target = 56803
输出:11361
提示:
- 1 <= arr.length <= 10^4
- 1 <= arr[i], target <= 10^5
2 题解
本题使用前缀和+二分搜索,先将数组排序并记录前缀和
p
r
e
f
i
x
prefix
prefix,规定二分查找的左边界为
0
0
0,右边界为数组最大值,二分查找要替换的值。我们需要计算替换值后数组的和,因此我们可以再使用一个二分查找在数组中搜索最后一个小于替换值的元素的下标,这样替换后数组的和就可以计算为:
s
=
p
r
e
f
i
x
[
i
n
d
e
x
]
+
m
i
d
∗
(
n
−
i
n
d
e
x
)
s = prefix[index] + mid*(n-index)
s=prefix[index]+mid∗(n−index)
其中
m
i
d
mid
mid为替换值,
i
n
d
e
x
index
index为最后一个小于替换值的元素的下标,
n
n
n为数组长度。
另外,这里查找替换值的二分查找退出条件使用的是 i < j − 1 i<j-1 i<j−1,最后要判断 i i i和 j j j到底哪一个是正确的返回值。
class Solution:
def findBestValue(self, arr: List[int], target: int) -> int:
arr.sort()
n = len(arr)
prefix = [0]
s = 0
for i in range(n):
s += arr[i]
prefix.append(s)
i, j = 0, arr[-1]
while i < j - 1:
mid = (i + j) // 2
index = bisect.bisect_left(arr, mid)
s = prefix[index] + mid*(n - index)
if s == target:
return mid
elif s > target:
j = mid
else:
i = mid
a = bisect.bisect_left(arr, i)
a = prefix[a] + i*(n - a)
b = bisect.bisect_left(arr, j)
b = prefix[b] + j*(n - b)
return i if abs(a - target) <= abs(b - target) else j