假设我们有一个称为堆的数字列表和一个值 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