刷遍LeetCode之33：搜索旋转排序数组

题目描述

假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如，数组 [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

需要进一步改进：

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],但是目标值却在左半段。