【LeetCode-数组】拥有最多糖果的孩子

题目描述

给你一个数组 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)