日渐头秃的代码日记 -- 9 LC第287周赛第三题

211 阅读1分钟

首先来吐槽一下,这道题就很烦。。。。思路都对,就是超时。。。以及,地球上哪有那么多小孩,出题人怎么想的。。。。

每个小孩最多能分到多少糖果

给你一个 下标从 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 <= 105
  • 1 <= candies[i] <= 107
  • 1 <= 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