题目描述:
假设一个单调递增的数组里的每个元素都是整数并且是唯一的。请编程实现一个函数找出数组中任意一个数值等于其下标的元素。例如,在数组[-3, -1, 1, 3, 5]中,数字3和它的下标相等。
样例:
输入:[-3, -1, 1, 3, 5]
输出:3
注意:如果不存在,则返回-1。
分析:
首先想到的方法是对数组从头遍历,当遍历到一个元素和其下标相等时直接返回,如果遍历结束仍没有找到则返回-1。这种方法的时间复杂度是O(n)。
第二种方法考虑输入数据的特点:单调递增且唯一,所以使用二分查找。
-
如果mid=nums[mid],则返回mid;
-
如果mid>nums[mid],则mid左侧的元素肯定不再匹配。
例如[-1,0,1,3,5]
-
如果mid<nums[mid],则mid右侧的元素肯定不再匹配。
例如[0,1,3,4,5]
class Solution(object):
def getNumberSameAsIndex(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
sta=0
end = len(nums)-1
while sta<=end:
mid = sta+(end-sta)//2
if mid == nums[mid]:
return mid
elif mid>nums[mid]:
sta = mid+1
else:
end = mid-1
return -1