Skip to content

1552.两球之间的磁力

binary search + greedy, https://leetcode.cn/problems/magnetic-force-between-two-balls/

在代号为 C-137 的地球上,Rick 发现如果他将两个球放在他新发明的篮子里,它们之间会形成特殊形式的磁力。Rick 有 n 个空的篮子,第 i 个篮子的位置在 position[i] ,Morty 想把 m 个球放到这些篮子里,使得任意两球间 最小磁力 最大。

已知两个球如果分别位于 xy ,那么它们之间的磁力为 |x - y|

给你一个整数数组 position 和一个整数 m ,请你返回最大化的最小磁力。

示例 1:

img
输入:position = [1,2,3,4,7], m = 3
输出:3
解释:将 3 个球分别放入位于 1,4 和 7 的三个篮子,两球间的磁力分别为 [3, 3, 6]。最小磁力为 3 。我们没办法让最小磁力大于 3 。

示例 2:

输入:position = [5,4,3,2,1,1000000000], m = 2
输出:999999999
解释:我们使用位于 1 和 1000000000 的篮子时最小磁力最大。

提示:

  • n == position.length
  • 2 <= n <= 10^5
  • 1 <= position[i] <= 10^9
  • 所有 position 中的整数 互不相同
  • 2 <= m <= position.length
python
from typing import List


class Solution:
    def maxDistance(self, position: List[int], m: int) -> int:
        def check(distance):
            count = 1
            pre = position[0]
            for fast in position[1:]:
                if fast - pre >= distance:
                    count += 1
                    pre = fast

            return count >= m

        position.sort()
        left, right = 0, position[-1] - position[0] + 1
        ans = -1
        while left < right:
            mid = (left + right) // 2
            if check(mid):
                ans = mid
                left = mid + 1
            else:
                right = mid
        return ans


if __name__ == "__main__":
    position = [1, 2, 3, 4, 7]
    m = 3
    print(Solution().maxDistance(position, m))