给你一个非负整数数组 nums 。如果存在一个数 x ,使得 nums 中恰好有 x 个元素 大于或者等于 x ,那么就称 nums 是一个 特殊数组 ,而 x 是该数组的 特征值 。
注意: x 不必 是 nums 的中的元素。
如果数组 nums 是一个 特殊数组 ,请返回它的特征值 x 。否则,返回 -1 。可以证明的是,如果 nums 是特殊数组,那么其特征值 x 是 唯一的 。
解法1:
将数组从大到小排列,在[1, nums.size()]范围内枚举x的值。
class Solution {
public:
int specialArray(vector<int>& nums) {
sort(nums.begin(), nums.end(), std::greater<int>());
size_t i = 1;
for (; i < nums.size(); ++i) {
if (nums[i - 1] >= i && nums[i] < i)
break;
}
if (i < nums.size() || nums.back() >= i)
return i;
return -1;
}
};
解法2:
计数。
class Solution {
public:
int specialArray(vector<int>& nums) {
vector<int> cnt(nums.size() + 1, 0);
for (int i : nums) {
if (i >= nums.size())
++cnt[nums.size()];
else
++cnt[i];
}
int sum = 0;
for (size_t i = cnt.size() - 1; i > 0; --i) {
sum += cnt[i];
if (sum == i)
return i;
}
return -1;
}
};
上述代码中的sum可以省略。
class Solution {
public:
int specialArray(vector<int>& nums) {
vector<int> cnt(nums.size() + 1, 0);
for (int i : nums) {
if (i >= nums.size())
++cnt[nums.size()];
else
++cnt[i];
}
for (size_t i = cnt.size() - 1; i > 0; --i) {
if (cnt[i] == i)
return i;
cnt[i - 1] += cnt[i];
}
return -1;
}
};