货仓选址问题(中位数贪心)
本文最后更新于 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
版权声明:除特殊说明,博客文章均为大块肌原创,依据CC BY-SA 4.0许可证进行授权,转载请附上出处链接及本声明。
暂无评论

发送评论 编辑评论


				
|´・ω・)ノ
ヾ(≧∇≦*)ゝ
(☆ω☆)
(╯‵□′)╯︵┴─┴
 ̄﹃ ̄
(/ω\)
∠( ᐛ 」∠)_
(๑•̀ㅁ•́ฅ)
→_→
୧(๑•̀⌄•́๑)૭
٩(ˊᗜˋ*)و
(ノ°ο°)ノ
(´இ皿இ`)
⌇●﹏●⌇
(ฅ´ω`ฅ)
(╯°A°)╯︵○○○
φ( ̄∇ ̄o)
ヾ(´・ ・`。)ノ"
( ง ᵒ̌皿ᵒ̌)ง⁼³₌₃
(ó﹏ò。)
Σ(っ °Д °;)っ
( ,,´・ω・)ノ"(´っω・`。)
╮(╯▽╰)╭
o(*////▽////*)q
>﹏<
( ๑´•ω•) "(ㆆᴗㆆ)
😂
😀
😅
😊
🙂
🙃
😌
😍
😘
😜
😝
😏
😒
🙄
😳
😡
😔
😫
😱
😭
💩
👻
🙌
🖕
👍
👫
👬
👭
🌚
🌝
🙈
💊
😶
🙏
🍦
🍉
😣
Source: github.com/k4yt3x/flowerhd
颜文字
Emoji
小恐龙
花!
上一篇
下一篇