假定有一个升序排序的有序数组,在某个位置进行了分组交换,保持交换后的循环方向不变,如0 1 2 4 5 6 7可能会变成 4 5 6 7 0 1 2,前后收尾可以连成同样的环,现有一个旋转后的数组,给定一个目标值,如何寻找他的位置呢?
分析:
在有序数组中我们可以使用二分法提高查找速度,但是这里存在一个问题。以456012为例,假如我们要寻找的目标值为1,而1在7012中,前后两端的元素值的大小关系难以确定,所以我们需要对目标值所处的区间进行划分。如果我们仔细分析就会发现,区间可以分为两类,一类是左端点的值大于右端点的值,一类是右端点值大于左端点的值,所以我们需要对这两种情况进行划分,分别进行讨论,代码如下:
def search(nums, target):
if not nums:
return -1
low, high = 0, len(nums) - 1
while low <= high:
mid = (low + high) / 2
if target == nums[mid]:
return mid
if nums[low] <= nums[mid]:
if nums[low] <= target <= nums[mid]:
high = mid - 1
else:
low = mid + 1
else:
if nums[mid] <= target <= nums[high]:
low = mid + 1
else:
high = mid - 1
return -1