(nice!!!)LeetCode 2860. 让所有学生保持开心的分组方法数 (排序、枚举)

题目:2860. 让所有学生保持开心的分组方法数

(nice!!!)LeetCode 2860. 让所有学生保持开心的分组方法数 (排序、枚举)_第1张图片
(nice!!!)LeetCode 2860. 让所有学生保持开心的分组方法数 (排序、枚举)_第2张图片
思路:先将数组nums进行升序排序。然后假设第i-1位学生选中,而第i位学生没有选中,并且所有学生都符合开心要求。
那么根据开心要求,[0,i-1]的学生,nums值都是小于被选中的人数k;且[i,n-1]的学生,nums值都是大于被选中的人数k。这时被选中的人数就是 i,即nums[i-1] 当nums[0]>0时,可以一个学生都不选。
当nums[n-1]

class Solution {
public:
    int countWays(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        int ans=0;
        //可以一个学生都不选
        if(nums[0]>0) ans++;
        //枚举第i-1位学生选中,而第i位学生没有选中
        for(int i=1;i<nums.size();i++){
        	//所有学生都符合开心要求
            if(nums[i-1]<i&&i<nums[i]) ans++;
        }
        //+1表示所有学生都选的情况
        return ans+1;
    }
};

你可能感兴趣的:(LeetCode,leetcode,算法,职场和发展,c++,暴力枚举,数据结构)