1552. 两球之间的磁力

在代号为 C-137 的地球上,Rick 发现如果他将两个球放在他新发明的篮子里,它们之间会形成特殊形式的磁力。Rick 有 n 个空的篮子,第 i 个篮子的位置在 position[i] ,Morty 想把 m 个球放到这些篮子里,使得任意两球间 最小磁力 最大。
已知两个球如果分别位于 xy ,那么它们之间的磁力为 |x - y|
给你一个整数数组 position 和一个整数 m ,请你返回最大化的最小磁力。
示例 1:
notion imagenotion image
输入: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

法 1 二分查找

思路
用二分法去猜测答案。
因为是找最小间距的最大值,所以相当于找右边界。至于判断当前尝试的间距答案是否合法,只需要计算一下以当前的间距放的话,能放几个,是否满足题目给的磁珠数目。
题解
class Solution: def maxDistance(self, position: List[int], m: int) -> int: position.sort() # left, right分别为最小磁力和最大磁力 left, right = 1, position[-1] - position[0] while left <= right: mid = left + (right - left) // 2 # 最小磁力为 mid 看是否可行 # 因为是要最大化,所以是找 右边界 if self.isValid(position, m, mid): left = mid + 1 else: right = mid - 1 return right def isValid(self, position, m, mid): # 有 m 个磁珠 # 最近位置为 mid # 所以计算一下以当前间距放的话 能放几个 count = 1 prev, cur = position[0], None for cur in position[1:]: if cur - prev >= mid: count += 1 prev = cur return count >= m