LSGO——LeetCode实战（数组系列）：33题 搜索旋转排序数组(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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：

思路：
对旋转后的两段有序数组，分别进行二分搜索。

程序:

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        if nums ==[]:
            return -1
        if len(nums) ==1:
            if target == nums[0]:
                return 0
            return -1 
        model = 1
        while nums[model-1] < nums[model] and model < len(nums)-1:
            model +=1
        if self.search_nums(0,model-1,nums,target)==-1:
            return self.search_nums(model,len(nums)-1,nums,target)
        else:
            return self.search_nums(0,model-1,nums,target)
            
    def search_nums(self,left,right,nums,target):
        while left <= right:
            model = (left+right) //2
            if nums[model] ==target:
                return model
            elif nums[model] >target:
                right = model-1
            else:
                left = model +1
        return -1

解法二：

思路：对整个数组使用二分搜索，由于不同于一般的有序数组，旋转数组的mid取值情况需要单独讨论。

• if nums[0] <= nums[i] 那么 nums[0] 到 nums[i] 为有序数组,那么当 nums[0] <= target <= nums[i] 时我们应该在 0-i范围内查找；

• if nums[i] < nums[0] 那么在 0-i0−i 区间的某个点处发生了下降（旋转），那么 i+1 到最后一个数字的区间为有序数组，并且所有的数字都是小于 nums[0] 且大于 nums[i]，当target不属于 nums[0] 到 nums[i] 时（target <= nums[i] < nums[0] or nums[i] < nums[0] <= target），我们应该在 0-i区间内查找。

程序一：

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        lo = 0
        hi = len(nums)-1
        while (lo < hi) :
            mid = (lo + hi) // 2
            if ((nums[0] > target) ^ (nums[0] > nums[mid]) ^ (target > nums[mid])):
                lo = mid + 1
            else:
                hi = mid
        
        return  lo if lo == hi and nums[lo] == target else -1

程序二：

思路：程序二相对于程序一，思路是一样，但是程序的实现相对易懂一些。

class Solution:
    def search(self, nums, target):
        size = len(nums)
        if size == 0:
            return -1

        left = 0
        right = size - 1
        while left < right:
            # mid = left + (right - left + 1) // 2
            mid = (left + right + 1) >> 1
            if nums[mid] < nums[right]:
                # [7,8,9,1,2,3,4,5,6] ，后半部分有序
                if nums[mid] <= target <= nums[right]:
                    left = mid
                else:
                    right = mid - 1
            else:
                # 注意：写这个 if 语句的时候，要让分支和上面一样
                # [4,5,6,7,8,9,0,1,2]，前半部分有序
                if nums[left] <= target <= nums[mid - 1]:
                    right = mid - 1
                else:
                    left = mid
        # 后处理
        return left if nums[left] == target else -1