题目描述
给你一个数组 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 个额外糖果,那么他将成为拥有最多糖果的孩子。
题目链接: https://leetcode-cn.com/problems/kids-with-the-greatest-number-of-candies/
思路
首先找到数组中的最大元素,记为 maxVal。然后遍历数组,判断 maxVal 减去当前元素 nums[i] 是否小于等于 extraCandies:如果小于等于,则说明 nums[i]+extraCandies ≥ maxVal,为 true;否则为 false。
代码如下:
class Solution {
public:
vector<bool> kidsWithCandies(vector<int>& candies, int extraCandies) {
if(candies.empty()) return {};
int maxVal = 0;
for(int i=0; i<candies.size(); i++){
maxVal = max(candies[i], maxVal);
}
vector<bool> ans;
for(int i=0; i<candies.size(); i++){
if(maxVal-candies[i]<=extraCandies){
ans.push_back(true);
}else ans.push_back(false);
}
return ans;
}
};
- 时间复杂度:O(n)
- 空间复杂度:o(n)