用Python求K小时去石率的程序

假设我们有一个称为堆的数字列表和一个值 k。桩 [i] 表示桩 i 上的石头数量。在每个小时,我们选择任何一堆并从该堆中移除 r 个石头。如果我们捡到少于 r 个石头的一堆,清理一堆仍然需要一个小时。我们必须找到 r 的最小值,以便我们可以在小于或等于 k 小时内移除所有石头。

所以,如果输入像堆 = [3, 6, 4] k = 5,那么输出将是 3,因为对于 r = 3 石头每小时,我们可以在 2 小时内清除第二堆,然后清除2 小时内第三次,1 小时内清除第一堆。

示例

让我们看看以下实现以获得更好的理解 -

from math import ceil
def solve(piles, k):
   l = 1
   h = max(piles)
   r = h

   def turns(r):
      return sum(ceil(b / r) for b in piles)
   while l < h:
      mid = (l + h) // 2
      if turns(mid) > k:
         l = mid + 1
      else:
         h = mid
         r = min(r, mid)
   return r

piles = [3, 6, 4]
k = 5
print(solve(piles, k))

输入

[3, 6, 4], 5
输出结果
3

猜你喜欢