【5分鐘力扣】1431. 擁有最多糖果的孩子

一、前言

兒童節快樂，新的一月來了，給自己這個大齡兒童最好的禮物就是刷題了！
每天五分鐘，看懂一道簡單、中等難度的算法題，儘可能將複雜的題講清楚。
瘋狂學習python中，2020-06-01更新

二、題目

給你一個數組 candies 和一個整數 extraCandies ，其中 candies[i] 代表第 i 個孩子擁有的糖果數目。

對每一個孩子，檢查是否存在一種方案，將額外的 extraCandies 個糖果分配給孩子們之後，此孩子有 最多 的糖果。注意，允許有多個孩子同時擁有 最多 的糖果數目。

示例 1：

輸入：candies = [2,3,5,1,3], extraCandies = 3
輸出：[true,true,true,false,true]

解釋：
孩子 1 有 2 個糖果，如果他得到所有額外的糖果（3個），那麼他總共有 5 個糖果，他將成爲擁有最多糖果的孩子。
孩子 2 有 3 個糖果，如果他得到至少 2 個額外糖果，那麼他將成爲擁有最多糖果的孩子。
孩子 3 有 5 個糖果，他已經是擁有最多糖果的孩子。
孩子 4 有 1 個糖果，即使他得到所有額外的糖果，他也只有 4 個糖果，無法成爲擁有糖果最多的孩子。
孩子 5 有 3 個糖果，如果他得到至少 2 個額外糖果，那麼他將成爲擁有最多糖果的孩子。

示例 2：

輸入：candies = [4,2,1,1,2], extraCandies = 1
輸出：[true,false,false,false,false]
解釋：只有 1 個額外糖果，所以不管額外糖果給誰，只有孩子 1 可以成爲擁有糖果最多的孩子。

示例 3：

輸入：candies = [12,1,12], extraCandies = 10
輸出：[true,false,true]

提示：

2 <= candies.length <= 100
1 <= candies[i] <= 100
1 <= extraCandies <= 50

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/kids-with-the-greatest-number-of-candies
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

三、解題思路

3.1 列舉法

先理解一下題意：

擁有最多糖果的人就是列表中，數值最大的那位，對於其他小朋友而言他們「擁有的糖果數目」加上「額外的糖果數目」> = 最多糖果的人，就滿足條件，輸出True，反之輸出False。

該題時間複雜度爲 ：O(n)，空間複雜度：O(1)

設某個小朋友的糖果數爲 x，其餘小朋友擁有的糖果數目最大值爲 y，額外的糖果數爲 e。這個小朋友可以擁有最多的糖果，當且僅當

$x+e >= y$

由於 x+e≥x 顯然成立，那麼我們有
$x+e≥max(x,y)$

四、代碼示例

粗糙的代碼

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
        mostCandy = max(candies)
        for idx, i in enumerate(candies):
            if i + extraCandies >= mostCandy:
                candies[idx] = True
            else:
                candies[idx] = False
        return candies
    

優化後的代碼

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
        mostCandy = max(candies)
        result = [False for i in range(len(candies))]
        for i in range(len(candies)):
            if candies[i] + extraCandies >= mostCandy:
                result[i] = True
        return result

極簡代碼

class Solution:
    def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
        mostCandy = max(candies)
        return [ (i+extraCandies)>=mostCandy for i in candies]