用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