leetcode2860. 让所有学生保持开心的分组方法数-medium-CSDN博客

本文链接：https://blog.csdn.net/weixin_37253733/article/details/148067448

1 题目：让所有学生保持开心的分组方法数

官方标定难度：中

给你一个下标从 0 开始、长度为 n 的整数数组 nums ，其中 n 是班级中学生的总数。班主任希望能够在让所有学生保持开心的情况下选出一组学生：

如果能够满足下述两个条件之一，则认为第 i 位学生将会保持开心：

这位学生被选中，并且被选中的学生人数严格大于 nums[i] 。
这位学生没有被选中，并且被选中的学生人数严格小于 nums[i] 。
返回能够满足让所有学生保持开心的分组方法的数目。

示例 1：

输入：nums = [1,1]
输出：2
解释：
有两种可行的方法：
班主任没有选中学生。
班主任选中所有学生形成一组。
如果班主任仅选中一个学生来完成分组，那么两个学生都无法保持开心。因此，仅存在两种可行的方法。

示例 2：

输入：nums = [6,0,3,3,6,7,2,7]
输出：3
解释：
存在三种可行的方法：
班主任选中下标为 1 的学生形成一组。
班主任选中下标为 1、2、3、6 的学生形成一组。
班主任选中所有学生形成一组。

提示：

$1 <= nums.length <= 10^5$
0 <= nums[i] < nums.length

2 solution

先排序，然后从小到达枚举

代码

class Solution {
    /*
     * 选择 m 个数，选中的数必须都 < m。剩余的数必须都大于 m
     * 从小到大开始选，枚举即可
     */
public:
    int countWays(vector<int> &nums) {
        int n = nums.size();
        vector<int> cnt(n);
        for (int i: nums) cnt[i]++;
        int s = 0, ans = 0;

        int j = -1;
        for (int i = 0; i < n; i++) {
            if (cnt[i]) { // 选择 < i 的数
                // cout << j << " " << s << " " << s << endl;
                if (j < s && s < i) ans++;
                j = i;
                s += cnt[i];
            }
        }
        return ans + 1;
    }
};