- 特殊数组的特征值
给你一个非负整数数组 nums 。如果存在一个数 x ,使得 nums 中恰好有 x 个元素 大于或者等于 x ,那么就称 nums 是一个 特殊数组 ,而 x 是该数组的 特征值 。
注意: x 不必 是 nums 的中的元素。
如果数组 nums 是一个 特殊数组 ,请返回它的特征值 x 。否则,返回 -1 。可以证明的是,如果 nums 是特殊数组,那么其特征值 x 是 唯一的 。
示例 1:
输入:nums = [3,5]
输出:2
解释:有 2 个元素(3 和 5)大于或等于 2 。
老实讲,就二分这个方面来说,我感觉这个题不应该是简单题。。。也可能是我把简单的问题复杂化了吧。
思路
特征值的定义已经get了,有一个数x,nums中恰好有x个数大于等于x。
因为要判断多少个数大于它,显然先把他排序好更方便操作。
对于每个元素,从它到末尾的元素就是大于等于它的元素的个数,再和这个元素的值进行比较。由于这个数不一定在数组之中,所以他就应该在第一个nums[i]超过比他大的元素的位置和nums[i-1]之间,对于这个,再采用一次二分。那就是找到第一个满足nums[i]>=n-i-1的值,要的结果就在nums[i]和nums[i-1]之间。
class Solution {
public:
int specialArray(vector<int>& nums) {
sort(nums.begin(),nums.end());
int n=nums.size();
int l=0,r=nums.size()-1;
//找到第一个满足nums[i]>=n-mid-1的位置
while(l<r)
{
int mid=l+(r-l)/2;
if(nums[mid]<n-mid-1)
{
l=mid+1;
}else
{
r=mid;
}
}
//结果一定在nums[l-1]和nums[l]之间,因为它可能不在数组中,所以不可以直接返回l
//考虑l为0的两种特殊情况(因为这里我们找不到nums[l-1])
if(l==0&&nums[0]==0)
{
return -1;
}
if(l==0)
{
return n-l;
}
int left=nums[l-1],right=nums[l];
while(left<=right)
{
int mid=left+(right-left)/2;
if(mid<n-l)
{
left=mid+1;
}else if(mid>n-l)
{
right=mid-1;
}else
{
return mid;
}
}
return -1;
}
};