首先来吐槽一下,这道题就很烦。。。。思路都对,就是超时。。。以及,地球上哪有那么多小孩,出题人怎么想的。。。。
每个小孩最多能分到多少糖果
给你一个 下标从 0 开始 的整数数组 candies 。数组中的每个元素表示大小为 candies[i] 的一堆糖果。你可以将每堆糖果分成任意数量的 子堆 ,但 无法 再将两堆合并到一起。
另给你一个整数 k 。你需要将这些糖果分配给 k 个小孩,使每个小孩分到 相同 数量的糖果。每个小孩可以拿走 至多一堆 糖果,有些糖果可能会不被分配。
返回每个小孩可以拿走的 最大糖果数目 **。
示例 1:
输入: candies = [5,8,6], k = 3
输出: 5
解释: 可以将 candies[1] 分成大小分别为 5 和 3 的两堆,然后把 candies[2] 分成大小分别为 5 和 1 的两堆。现在就有五堆大小分别为 5、5、3、5 和 1 的糖果。可以把 3 堆大小为 5 的糖果分给 3 个小孩。可以证明无法让每个小孩得到超过 5 颗糖果。
示例 2:
输入: candies = [2,5], k = 11
输出: 0
解释: 总共有 11 个小孩,但只有 7 颗糖果,但如果要分配糖果的话,必须保证每个小孩至少能得到 1 颗糖果。因此,最后每个小孩都没有得到糖果,答案是 0 。
提示:
1 <= candies.length <= 1051 <= candies[i] <= 1071 <= k <= 1012
基本思路就是从最大可能的数量开始尝试,直到找到刚好能够分完的情况。 最大可能数字就是所有糖果一个不剩地平均分完,最坏情况一人一个,只不过是从最大到1的遍历过程中,需要点加速技巧,使用二分法,就这么一个考点。
from math import floor, ceil
class Solution:
def maximumCandies(self, candies: List[int], k: int) -> int:
self.candies = candies
self.k = k
total = sum(candies)
if total < k:
return 0
r = floor(total/k)
l = 1
if self.check_m(r):
return r
while (l < r-1):
m = int((l + r) / 2)
print(l, r, m)
if self.check_m(m):
l = m
else:
r = m
return l
def check_m(self, m):
x = 0
for c in self.candies:
x += floor(c/m)
if x >= self.k:
return True
else:
return False