1. 题目来源
题单:
- 思维
- 01.1、思维
2. 题目解析
有一定的思考难度,不太好归类,就放到 思维 里面吧。
思路:
- 首先分析题目中的一些关键信息,能得出来,选法是唯一的,且选的话,会把小于当前值的前面部分全部选了。
- 那么此时问题就转化为 选0、选1、选2…选n 个的问题了。
- 判断条件此时也很简单。
- 选 k 个,那么选中的最大值需要严格小于 k,未选中的最小值需要严格大于 k 即可。
具体证明,看靠前题解即可。描述起来有点困难,不赘述。
代码实现上需要注意下,不需要将 未选择、全选,当做特殊情况处理,一开始直接根据下标考虑,直接从下标 0 开始判断,且最终直接来了个 i+1 的判断逻辑,遗漏了这两种情况。
官解中根据划分长度来考虑会更加合适,即:枚举分组长度为 0、1、2、3、…n 这样子更加合适。包含了这两种特殊的情况。
学习下特判少的写法,更不易出错。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( 1 ) O(1) O(1)
一开始的写法
class Solution {
public:
int countWays(vector<int>& nums) {
int n = nums.size();
sort(nums.begin(), nums.end());
int res = 0;
for (int i = 0; i < n; i ++ ) {
int cnt = i + 1;
bool flag1 = false, flag2 = false;
if (cnt > nums[i]) flag1 = true;
if (i + 1 < n && nums[i + 1] > cnt) flag2 = true;
if (flag1 && flag2) res ++ ;
}
// 未选中
if (nums[0] > 0) res ++;
// 全选中
if (nums.back() < n) res ++ ;
return res;
}
};
简洁写法:
class Solution {
public:
int countWays(vector<int>& nums) {
int n = nums.size();
sort(nums.begin(), nums.end());
int res = 0;
for (int k = 0; k <= n; k ++ ) {
if (k > 0 && nums[k - 1] >= k) continue; // k=n,n个全选,判断最大值是否小于k个
if (k < n && nums[k] <= k) continue; // k=0,0个全不选,判断最小值是否大于k
res ++ ;
}
return res;
}
};