本文最后更新于 282 天前,如有失效请评论区留言。
货仓选址问题常用来求在几个数(一个数组)值中间选一个数,使得这个数到数组所有数的距离之和代价最小。这时选取中间的数是最优的想法(奇数中位数,偶数[nums[(n-1)//2], nums[n//2]]范围内都可以),下面是两种代码模板:
滑窗+前缀和
def maxFrequencyScore(self, nums: List[int], k: int) -> int:
nums.sort()
n = len(nums)
s = list(accumulate(nums, initial=0)) # nums 的前缀和
# 把 nums[l] 到 nums[r] 都变成 nums[i]
def distance_sum(l: int, i: int, r: int) -> int:
left = nums[i] * (i - l) - (s[i] - s[l])
right = s[r + 1] - s[i + 1] - nums[i] * (r - i)
return left + right
ans = left = 0
for i in range(n):
while distance_sum(left, (left + i) // 2, i) > k:
left += 1
ans = max(ans, i - left + 1)
return ans
滑窗+贡献法
def maxFrequencyScore(self, nums: List[int], k: int) -> int:
nums.sort()
ans = left = s = 0 # s 是窗口元素与窗口中位数的差之和
for right, x in enumerate(nums):
s += x - nums[(left + right) // 2]
while s > k:
s += nums[left] - nums[(left + right + 1) // 2]
left += 1
ans = max(ans, right - left + 1)
return ans