1. 程式人生 > 實用技巧 >1431. 擁有最多糖果的孩子

1431. 擁有最多糖果的孩子

難度簡單

給你一個數組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 {
public:
    vector<bool> kidsWithCandies(vector<int>& candies, int extraCandies) {
        if (candies.empty()) {
            return vector<bool>();
        }
        vector<bool> res(candies.size(), false);
        int max_val = INT_MIN;
        for (auto val : candies) {
            if (val > max_val) {
                max_val = val;
            }
        }
        for (auto i = 0; i < candies.size(); ++i) {
            if (candies[i] + extraCandies >= max_val) {
                res[i] = true;
            }
        }
        return res;
    }
};