1.题目描述
给你一个非负整数数组
nums
。如果存在一个数x
,使得nums
中恰好有x
个元素 大于或者等于x
,那么就称nums
是一个 特殊数组 ,而x
是该数组的 特征值 。注意:
x
不必 是nums
的中的元素。如果数组
nums
是一个 特殊数组 ,请返回它的特征值x
。否则,返回-1
。可以证明的是,如果nums
是特殊数组,那么其特征值x
是 唯一的
输入:nums = [3,5]
输出:2
解释:有 2 个元素(3 和 5)大于或等于 2 。
输入:nums = [0,0]
输出:-1
解释:没有满足题目要求的特殊数组,故而也不存在特征值 x 。
如果 x = 0,应该有 0 个元素 >= x,但实际有 2 个。
如果 x = 1,应该有 1 个元素 >= x,但实际有 0 个。
如果 x = 2,应该有 2 个元素 >= x,但实际有 0 个。
x 不能取更大的值,因为 nums 中只有两个元素。
输入:nums = [0,4,3,0,4]
输出:3
解释:有 3 个元素大于或等于 3 。
输入:nums = [3,6,7,7,0]
输出:-1
提示:
1 <= nums.length <= 100
0 <= nums[i] <= 1000
2.思路分析
2.1 模拟
根据题目描述, x x x的取值范围应该是 [ 1 , n u m s . l e n g t h ] [1, nums.length] [1,nums.length],那么为了便于判断比对,我们首先需要将nums进行升序排序。然后,根据下图的例子,我们可以得出如下结论:
- 当 x x x等于 n u m s . l e n g t h nums.length nums.length 时,需要满足 n u m s [ 0 ] > = n u m s . l e n g t h nums[0] >= nums.length nums[0]>=nums.length;
- 当
x
x
x等于
[
1
,
n
u
m
s
.
l
e
n
g
t
h
−
1
]
[1, nums.length - 1]
[1,nums.length−1] 时,需要满足
n
u
m
s
[
i
]
>
=
(
r
e
s
=
n
u
m
s
.
l
e
n
g
t
h
−
i
)
nums[i] >= (res = nums.length - i)
nums[i]>=(res=nums.length−i) 并且
n
u
m
s
[
i
−
1
]
<
r
e
s
nums[i - 1] < res
nums[i−1]<res;
由于特征值x是唯一的,所以,只要有x满足了上面的判断条件,就将x值返回即可。如果没有符合的x,则返回-1;
2.2 二分查找
在 [1, nums.length]之间二分查找,注意:不是找 nums 内元素,只是找从 1 到 nums.length 的数是否满足
- 记当前数为 target
- 如果 target 等于 nums内大于或等于 target 的数量,即满足,则返回 target
- 如果 target 大于 nums 内大于或等于 target 的数量,就是 nums 内没有这么多大于 target 的元素,则收缩右边界
如果 target 小于 nums 内大于或等于 target 的数量,就是 nums 内太多大于 target 的元素,则收缩左边界
3.代码实现
3.1 模拟
class Solution:
def specialArray(self, nums: List[int]) -> int:
# 排序
nums.sort()
if nums[0] >= len(nums):
return len(nums)
for i in range(1, len(nums)):
if nums[i] >= len(nums) - i and nums[i - 1] < len(nums) - i:
return len(nums) - i
return -1
复杂度分析:
- 时间复杂度:O(nlogn),其中 n 为数组的长度。
- 空间复杂度:O(logn),即为排序需要的栈空间。
3.2 二分查找
class Solution:
def specialArray(self, nums: List[int]) -> int:
n = len(nums)
nums.sort()
left, right = -1, n + 1
while left <= right:
mid = (left + right) >> 1
if (t := n - bisect.bisect_left(nums, mid)) == mid:
return mid
elif t < mid:
right = mid - 1
else:
left = mid + 1
return -1
"""
bisect.bisect_left(a,x,lo=0,hi=len(a),*,key=None),在有序数组a中[lo,hi]区间内查找x插入的位置,返回的是索引值。如果a中有跟x相同的元素,则x插入的位置是左边,key指定了一个单参数的方法,该方法的返回值作为与k比较的基准。
"""
复杂度分析:
- 时间复杂度:O(nlogn),其中 n 为数组的长度。
- 空间复杂度:O(logn),即为排序需要的栈空间。