1608. 特殊数组的特征值

给你一个非负整数数组 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;
    }
};