题目:
有一个有序的int类型的数组arr,其中不含有重复元素,请找到满足arr[i]==i条件的最左的位置。如果所有位置上的数都不满足条件,返回-1。
给定有序数组arr及它的大小n,请返回所求值。
测试用例:
[-1,0,2,3],4返回:2
思路:
首先需要明确的是:虽然数组中的数是递增的,但是元素的序号是固定从0到n-1的。基于这一点,当arr[0] > 0的时候,由于arr[1] > a[0] > 0,∴arr[1] > 1。以此类推,arr中的每一个数都不可能有arr[i] == i。同理,当 arr[n-1] < n-1 时也是如此。
排除掉以上的特殊情形后,假设我们查找的区间是[0 : n),那么对于一个中间值arr[mid],如果arr[mid] < mid,那么arr[mid-1] < arr[mid] < mid, ∴ arr[mid-1] < mid-1,以此类推,所有在mid左边的元素都不可能是我们要找的答案,left = mid+1。同理,对于arr[mid] > mid 时,所有在mid右边的元素都不可能是要找的答案,right = mid(注意保持区间开闭性一致)。对于arr[mid] == mid的情形,mid可能是最终答案,但是由于我们要找的是最左原位,那么还应该到mid左边去寻找,right = mid。
代码:
# -*- coding:utf-8 -*-
class Find:
def findPos(self, arr, n):
# write code here
if arr[0] > 0 or arr[-1] < n-1:
return -1
left, right = 0, n
while left < right:
mid = left + (right-left)//2
if arr[mid] < mid:
left = mid+1
elif arr[mid] >= mid:
right = mid
if arr[right] == right:
return right
return -1
note:
这里在循环结束时,并没有判断是否会出现right == n,那么会出现这种情况吗?根据循环中的分支,想要right == n,那么每一次循环,arr[mid] < mid。由于代码开始时已经对特殊情况做了判断,所以我们可以构造一个数组,其arr[0] <= 0, arr[n-1] >= n-1,假设在[0 : n-1)区间内的元素都有arr[mid] < mid,那么当mid = n-1时,还是会使用第二个分支,即arr[mid] >= mid。这样一来,循环结束后,right不可能等于n