题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
解题思路:
假设原数组为[0,1,2,3,4,5,6,7]
,旋转后的数组为[4,5,6,7,0,1,2,3]
,目标值为target = 1
。
- 首先,由于算法的时间复杂度为对数时间,所以排除采用暴力的遍历查找法;
- 对于排序好的、无重复的数组进行搜索,考虑使用“二分查找”算法,而二分查找的关键是如何根据middle值来确定搜索的区间,即判断
a[mid] =?target
中间变量索引的值与目标值的大小关系,来确定下一步搜索区间的上界/下界; - 由于数组中元素的大小相对于其位置进行了改变,此时我们可以发现,middle索引出来的值
a[mid]=7 != target
;那么如何来规划下一步的搜索区间,使得target
的值位于正确的搜索区间中呢? - 我们发现,
mid
划分出来的数组的两部分,总有一部分是有序的(比如,[4,5,6,7]
),我们可以以此为标志来对下一步tartget
所在的区间进行确定: - 如果左半部分是有序数组
[low, mid]
且target
值位于该区间内,则使搜索的上界调整为high = mid - 1
;否则在[mid + 1, high
]区间内继续搜索; - 如果右半部分是有序数组
[mid, high]
且target
值位于该区间内,则使搜索的上界调整为low = mid + 1
;否则在[low, mid - 1
]区间内继续搜索;
代码块:
from typing import List
def search(nums: List[int], target: int) -> int: # python3新特性:类型注解
if len(nums) == 1:
if nums[0] == target:
return 0
else:
return -1
low, high = 0, len(nums)-1
while low<=high:
mid = low + ((high-low)>>1)
if nums[mid] == target:
return mid
if nums[low] <= nums[mid]: # 如果mid划分出来的左半边数组是有序的,则进一步判断target是否在此有序数组内
if nums[low]<= target <=nums[mid]: # 边界条件非常重要,因为若target值刚好出于数组边界,则有可能产生误判
high = mid - 1
else:
low = mid + 1
else:
if nums[mid+1] <= target <= nums[high]:
low = mid + 1
else:
high = mid - 1
return -1
需要进一步改进:
- 算法的空间复杂度还可以更加优化;
对于旋转数组nums = [4,5,6,7,0,1,2]
首先根据 nums[0]
与target
的关系判断target
是在左段还是右段。
例如 target = 5
, 目标值在左半段,因此在[4, 5, 6, 7, inf, inf, inf]
这个有序数组里找就行了;
例如 target = 1
, 目标值在右半段,因此在[-inf, -inf, -inf, -inf, 0, 1, 2]
这个有序数组里找就行了。
如此,我们又双叒叕将「旋转数组中找目标值」 转化成了 「有序数组中找目标值】
但是,这个思路不一定对,举一个反例:
假设nums = [8,9,0,1,2,3,4,5,6]
,target = 0
,尽管target < nums[0]
,但是目标值却在左半段。