题目:2860. 让所有学生保持开心的分组方法数
思路:先将数组nums进行升序排序。然后假设第i-1位学生选中,而第i位学生没有选中,并且所有学生都符合开心要求。
那么根据开心要求,[0,i-1]的学生,nums值都是小于被选中的人数k;且[i,n-1]的学生,nums值都是大于被选中的人数k。这时被选中的人数就是 i,即nums[i-1]<i<nums[i]。
当nums[0]>0时,可以一个学生都不选。
当nums[n-1]<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;
}
};