二分查找
这题如果是正常的有序数组的话可以直接用二分查找来解决。现在问题出在这个有序数组是被旋转过的,什么意思?也就是说从[1,2,3,4,5,6,7]
旋转成了[5,6,7,1,2,3,4]
,变成非有序的了。这种情况下,问题就会变得稍微复杂了些,但还是能通过二分查找来解决。
- 我们先观察下这种情况:
[5,6,7,1,2,3,4]
,mid为1。[low, mid-1]
是有序的,[mid, high]
是无序的。我们只需先判断target
在不在有序的那一侧,如果在就在[low, mid-1]
中寻找,否则就在[mid + 1, high]
中寻找。 - 反过来也是,如:
[6,7,1,2,3,4,5]
,mid为2。此时[mid, high]
是有序的,[low, mid-1]
是无序的,先判断target
在不在右侧,如果在就在[mid, high]
中找,否则就在[low, mid-1]
中找。
代码
class Solution:
def search(self, nums: List[int], target: int) -> int:
N = len(nums)
low, high = 0, N - 1
while low <= high:
mid = low + (high - low) // 2
if nums[mid] == target: return mid
if nums[0] <= nums[mid]:
if nums[0] <= target < nums[mid]:
high = mid - 1
else:
low = mid + 1
else:
if nums[mid] < target <= nums[N-1]:
low = mid + 1
else:
high = mid - 1
return -1