题目

给你一个数组 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 个额外糖果,那么他将成为拥有最多糖果的孩子。

思路

枚举法。找出最大糖果数,遍历一遍。
最大时间复杂度:【LeetCode】1431. 拥有最多糖果的孩子 - 图1

代码

两种代码,第一种很pythonic。

  1. class Solution:
  2. def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
  3. max_value = max(candies)
  4. return [True if (candy + extraCandies) >= max_value else False for candy in candies]

复习一下列表推导式的两种形式:
image.png

  1. class Solution:
  2. def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
  3. max_value = max(candies)
  4. ans = []
  5. for candy in candies:
  6. if candy + extraCandies >= max_value:
  7. ans.append(True)
  8. else:
  9. ans.append(False)
  10. return ans