前往此题

二分查找

这题如果是正常的有序数组的话可以直接用二分查找来解决。现在问题出在这个有序数组是被旋转过的,什么意思?也就是说从[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