1431. 拥有最多糖果的孩子(C++)

题目

给你一个数组 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 可以成为拥有糖果最多的孩子。

提示:

  • 2 <= candies.length <= 100
  • 1 <= candies[i] <= 100
  • 1 <= extraCandies <= 50

分析与题解

如果希望某个小朋友拥有的糖果最多,那么最优的方案是把额外的所有糖果都分给这个小朋友。

因此,我们可以枚举每一个小朋友,并将额外的所有糖果都分给这个小朋友,然后再用 O(n) 的时间遍历其余的小朋友,就可以判断这个小朋友是否拥有最多的糖果。

上述方法的时间复杂度为 O(n^2),然而我们可以将其优化为 O(n)。

事实上,对于每一个小朋友,只要这个小朋友拥有的糖果数目加上额外的糖果数目大于等于所有小朋友拥有的糖果数目最大值,那么这个小朋友就可以拥有最多的糖果。因此我们可以预处理出这个值,随后枚举每一个小朋友,只要这个小朋友拥有的糖果数目加上额外的糖果数目大于等于这个值,就可以满足要求。

代码如下:

class Solution {
public:
    vector<bool> kidsWithCandies(vector<int>& candies, int extraCandies) {
        int num = candies.size();
        int max = *max_element(candies.begin(),candies.end());
        vector<bool> res;
        for(int i=0;i<num;i++)
        {
            bool tmp = candies[i] + extraCandies >= max;
            res.push_back(tmp);
        }
        return res;
    }
};
posted @ 2020-08-12 00:05  脱线森林`  阅读(194)  评论(0编辑  收藏  举报