【LeetCode】33 & 81.搜索旋转排序数组 (Python)

33.搜索排序数组

https://leetcode-cn.com/problems/search-in-rotated-sorted-array/

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值，如果数组中存在这个目标值，则返回它的索引，否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

示例 1:

输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4
示例 2:

输入: nums = [4,5,6,7,0,1,2], target = 3
输出: -1

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        if not nums:
            return -1
        low, high = 0, len(nums) - 1
        while low < high:
            mid = (low + high) >> 1
            if nums[mid] == target:
                return mid
            if nums[mid] < target < nums[low] or nums[mid] >= nums[low] and (target < nums[low] or target > nums[mid]):
                low = mid + 1
            else:
                high = mid
                
        return low if nums[low] == target else -1

思路：要求logn时间复杂度，说明用二分法解决，也就是需要low, mid, high三个索引进行转换，我们可以考虑哪些情况需要将mid+1赋给low，首先可以肯定target在mid右边，那么一共有三种情况

1. target和mid都在旋转点右边，那么此时mums[mid] < target < nums[low]
2. target 在旋转点右边，mid在旋转点左边，此时nums[mid] > nums[low] and target < nums[low]
3. target和mid都在旋转点左边，此时nums[low] < nums[mid] < target

其他情况都是更新high为mid的。

 

81.搜索排序数组二

https://leetcode-cn.com/problems/search-in-rotated-sorted-array-ii/

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。

编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true，否则返回 false。

示例 1:

输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
示例 2:

输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false

class Solution:
    def search(self, nums: List[int], target: int) -> bool:
        if not nums:
            return False
        low, high = 0, len(nums) - 1
        while low < high:
            mid = (low + high) >> 1
            if nums[mid] == target:
                return True
            if nums[low] > target > nums[mid] or nums[mid] > nums[low] and (target < nums[low] or target > nums[mid]):
                low = mid +1
            elif nums[low] == nums[mid]:  ############
                low += 1
            else:
                high = mid
                
        return True if nums[low]==target else False

进阶版的列表有重复的地方，可以看到我们的代码只是添加了一行，就是当nums[low]和nums[mid]相等时，我们无法判断mid在旋转点前还是后，因此我们直接low += 1，跳过这个low，进行下一次的判断即可。