拥有最多糖果的孩子
给你一个数组
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
class Solution:
def kidsWithCandies(self, candies: List[int], extraCandies: int) -> List[bool]:
max_num = max(candies)
res = [candies[i] + extraCandies >= max_num for i in range(len(candies))]#列表推导式适用于数据小的情况,数据庞大时适用迭代器节省内存空间
#candies[i] + extraCandies >= max_num返回一个布尔值
return list(res)
-
时间复杂度:
这个函数的时间复杂度是O(n),其中n是
candies
列表的长度。首先,函数中的
max()
函数调用会遍历整个candies
列表,以找到最大值。这一步需要O(n)的时间复杂度。接下来,使用列表推导式创建结果列表
res
,会对candies
列表中的每个元素进行遍历,进行判断和运算。这一步同样需要O(n)的时间复杂度。因此,整个函数的时间复杂度是O(n)。无论
candies
列表的大小如何增长,函数的运行时间都会以线性的方式增长。 -
空间复杂度:
这个函数的空间复杂度是O(n),其中n是
candies
列表的长度。在函数中,创建了一个新的列表
res
来存储结果。这个列表的长度与candies
列表的长度相同,因此需要占用O(n)的额外空间。除此之外,其他的变量和操作都是常数级别的,不会随着输入大小的增加而增加额外的空间。
因此,整个函数的空间复杂度主要由结果列表
res
所占用的空间决定,是O(n)。