用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