【二分查找】leetcode 1608. 特殊数组的特征值

1608. 特殊数组的特征值

题目描述

给你一个非负整数数组 nums 。如果存在一个数 x ，使得 nums 中恰好有 x 个元素 大于或者等于 x ，那么就称 nums 是一个 特殊数组 ，而 x 是该数组的 特征值 。

注意： x 不必 是 nums 的中的元素。

如果数组 nums 是一个 特殊数组 ，请返回它的特征值 x 。否则，返回 -1 。可以证明的是，如果 nums 是特殊数组，那么其特征值 x 是 唯一的 。

示例1：

输入： nums = [3,5]
输出： 2
解释： 有 2 个元素（3 和 5）大于或等于 2 。

示例2：

输入： nums = [0,0]
输出： -1
解释： 没有满足题目要求的特殊数组，故而也不存在特征值 x 。
如果 x = 0，应该有 0 个元素 >= x，但实际有 2 个。
如果 x = 1，应该有 1 个元素 >= x，但实际有 0 个。
如果 x = 2，应该有 2 个元素 >= x，但实际有 0 个。
x 不能取更大的值，因为 nums 中只有两个元素。

示例3：

输入： nums = [0,4,3,0,4]
输出： 3
解释： 有 3 个元素大于或等于 3 。

示例4：

输入： nums = [3,6,7,7,0]
输出： -1

提示

• 1 <= nums.length <= 100
• 0 <= nums[i] <= 1000

方法：二分查找

解题思路

ans 初始值为 -1 ，说明数组中不存在特征值。
首先对数组排序，然后枚举所有可能会是特征值的的值。
最小的特征值是 1 ，最大的特征值是数组中最后一个元素。
然后在枚举每个可能是特征值的值的同时，在原数组进行二分查找，找到第一个 >= x 的元素的位置，判断是否满足条件 (有 x 个元素 >= x) 。如果满足条件，则更新 ans 。

代码

class Solution {
public:
    int specialArray(vector<int>& nums) {
        int ans = -1;
        sort(nums.begin(), nums.end());
        int x = 1;
        // 枚举所有的可能特征值
        while(x <= nums[nums.size() - 1])
        {
            
            int l = 0, r = nums.size(), mid;
            // 查找数组中第一个 >= x 的元素
            while(l < r)
            {
                mid = (l + r) >> 1;
                if(nums[mid] >= x)  r = mid;
                else    l = mid + 1;
            }
            // 判断是否满足条件，有 x 个元素 >= x
            if(nums.size() - l == x)
            {
                ans = x;
                break;
            }
            x++;
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度：O($nlogn$)。
• 空间复杂度：O(1)。