题目描述
给你一个数组 candies
和一个整数 extraCandies
,其中 candies[i]
代表第 i
个孩子拥有的糖果数目。
对每一个孩子,检查是否存在一种方案,将额外的 extraCandies
个糖果分配给孩子们之后,此孩子有 最多 的糖果。注意,允许有多个孩子同时拥有 最多 的糖果数目。
样例
输入: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 个额外糖果,那么他将成为拥有最多糖果的孩子。
输入:candies = [4,2,1,1,2], extraCandies = 1
输出:[true,false,false,false,false]
解释:只有 1 个额外糖果,所以不管额外糖果给谁,只有孩子 1 可以成为拥有糖果最多的孩子。
输入:candies = [12,1,12], extraCandies = 10
输出:[true,false,true]
限制
2 <= candies.length <= 100
1 <= candies[i] <= 100
1 <= extraCandies <= 50
算法
(贪心) $O(n)$
- 求出当前拥有糖果的人的最大值。
- 枚举每个人,将额外糖果全部给他,判断其与之前最大值的关系。
时间复杂度
- 遍历数组两次,时间复杂度为 $O(n)$。
空间复杂度
- 需要额外 $O(n)$ 的空间存储答案。
C++ 代码
class Solution {
public:
vector<bool> kidsWithCandies(vector<int>& candies, int extraCandies) {
int n = candies.size();
int ma = candies[0];
for (int i = 1; i < n; i++)
ma = max(ma, candies[i]);
vector<bool> ans(n, false);
for (int i = 0; i < n; i++)
if (candies[i] + extraCandies >= ma)
ans[i] = true;
return ans;
}
};