题目:假设一个单调递增的数组里的每个元素都是整数并且是唯一的。请编程实现一个函数找出数组中任意一个数值等于其下标的元素。例如,在数组[-3, -1, 1, 3, 5]中,数字3和它的下标相等。
输入:[-3, -1, 1, 3, 5]
输出:3
注意:如果不存在,则返回-1。
思路:二分
代码:
class Solution:
def func(self , nums):
if len(nums) == 0:
return -1
lo , hi = 0 , len(nums)-1
while lo < hi:
mid = (lo + hi) // 2
if nums[mid] < mid:
lo = mid + 1
elif nums[mid] > mid:
hi = mid -1
else:
return mid
return -1
a = [-3, -1, 1, 3, 5]
s = Solution()
print(s.func(a))