Leetcode【二分查找】| 33. 搜索旋转排序数组

题目

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

解题

思路

 采用的其实是二分查找的思想。

二分查找，又称为折半查找，是对有序表进行查找的方法。
因为查找的是有序的数组中某个元素target，每次将数组分成一半，根据两部分的边缘来判定target在哪一部分，从而每次都将查找范围减少一半。直到将目标范围缩小至1，找到target所在。
eg: 在有序数组nums = [1,2,3,4,5,6,7,8]中查找数字“3”所在位置。
　 ① 目标范围对半分[1,2,3,4]和[5,6,7,8]，通过比较target与边缘元素nums[mid]大小(3<4)，可判断3在[1,2,3,4]这半部分中，将目标范围从[1,2,3,4,5,6,7,8]缩小至[1,2,3,4]。
　 ② 将目标范围[1,2,3,4]再次对半分[1,2]和[3,4]，经过判断target = 3在[3,4]此范围中，将目标范围缩小至[3,4]
　 ③ 将目标范围[3,4]对半分为[3]和[4]，经过对比判断出target = 3在[3]中，找到target所在位置，查找结束

时间复杂度：O(logn)
空间复杂度：O(1)

二分查找的java实现：

public int search(int[] nums, int target) {
    int left = 0, right = nums.length - 1, mid = 0;
    while (left <= right) {
    	//mid = left +(right-left)/2 寻找中位数
        mid = lo + ((hi - lo) >> 1);
        if (nums[mid] == target) {
            return mid;
        }
        if (nums[mid] < target) {//目标在右半部分，范围缩小至右半部分
            left = mid + 1;
        } else {//目标在左半部分，范围缩小至左半部分
            right = mid - 1;
        }
    }
    return -1;
}

直接二分查找法

递归实现

按照二分查找的思想，根据边缘情况判断出不同情况下的缩小后的目标范围是哪半部分,然后递归。因为是旋转数组，不是单纯地有序，所以判断缩小后的目标范围没有有序数组那么简单，需要分情况判断（有一半是有序的，有一半还是旋转的）。

class Solution {
    public int search(int[] nums, int target) {
        //每次找一半 看哪一半是顺序的，哪一半是非顺序的，根据边界范围将目标范围缩小一半，直到只剩最后两半。
        int n = nums.length;
        if(n == 0) return -1;
        if(n == 1) return nums[0]==target?0:-1;
        return recursive(nums,0,n-1,target);
    }
    public int recursive(int[] nums, int left, int right,int target){
        if(right-left == 0){
            return nums[left] == target?0:-1;
        }
        int res = -1;
        int mid = (left+right)/2;

        if(target == nums[left]){
            return left;
        }else if(target == nums[right]){
            return right;
        }else if(target == nums[mid]){
            return mid;
        }
        //前半部分是顺序的 后半部分翻转 默认nums[left] >= nums[right]
        if(nums[left]<nums[mid]){
            if(target > nums[left] && target < nums[mid]){//表明target在左半边内
                res = recursive(nums,left,mid,target);
            }else if(target > nums[mid] || target < nums[right]){//target在右半边翻转前部分
                res = recursive(nums,mid+1,right,target);
            }else if(target < nums[left] && target > nums[right]){//target在翻转的两点中间值 说明不存在
                return -1;
            }
        }else{//翻转点在前半部分 后部部分顺序
            if(target > nums[mid] && target < nums[right]){//target在右半边顺序的部分
                res = recursive(nums,mid+1,right,target);
            }else if(target > nums[left] || target < nums[mid]){//target在左半边
                res = recursive(nums,left,mid,target);
            }else if(target < nums[left] && target > nums[right]){
                return -1;
            }
        }
        return res;
    }
}

非递归实现

用while()循环和left和right不断地更新来替代递归实现

class Solution {
   public int search(int[] nums, int target) {
    int left = 0, right = nums.length - 1, mid = 0;
    //用while循环代替递归 不停更新left指针和right指针位置
    while (left <= right) {
        mid = left + (right - left)/2;
        if (nums[mid] == target) {
            return mid;
        }
        // 先根据 nums[mid] 与 nums[left] 的关系判断 mid在左边有序还是右边有序数组中，即判断当前哪半部分时有序的，哪半部分时旋转的 
        if (nums[mid] >= nums[left]) {//左半边是顺序的
            // 再判断 target 是在 mid 的左边还是右边，从而调整左右边界 left 和 right
            if (target >= nums[left] && target < nums[mid]) {//target在左半边
                right = mid - 1;
            } else {//target在右半边
                left = mid + 1;
            }
        } else {//右半边是顺序的
            if (target > nums[mid] && target <= nums[right]) {
                left = mid + 1;
            } else {
                right = mid - 1;
            }
        }
    }
    return -1;
}
}

将旋转数组的二分查找变为有序数组的二分查找

思路参考Sweetiee: 多思路完全攻略
因为旋转数组可以分为两部分的有序数组，比如[4,5,6,7,0,1,2] 是由[4,5,6,7]和[0,1,2]这左右两个有序数组组成。我们将原问题分解为在这两个有序数组中的其中一个里查找target。同样是通过每次对半分来判断target在哪个有序数组中。
eg：nums = [4,5,6,7,0,1,2]，左边有序子数组[4,5,6,7]，右边有序子数组[0,1,2]

1. target = 0时，目标在右边有序子数组，将左边的都设置为inf，[inf,inf,inf,inf,0,1,2]，问题变为在有序数组中二分查找。
2. target = 6时，目标在左边有序子数组，将右边的有序子数组都设置为inf，[4,5,6,7,inf,inf,inf]，问题变为在有序数组中二分查找。

怎么设置inf ? 同样通过 每次对半分，判断mid在左边有序子数组还是右边有序子数组，根据情况依次设置mid为inf：

1. 当目标在右边子数组时，判断出当前mid在左边子数组，就将mid设置为inf;
2. 当目标在左边子数组时，判断出当前mid在右边子数组，就将mid设置为inf;

class Solution {
    public int search(int[] nums, int target) {
        int left = 0, right = nums.length - 1;
        while (left <= right) {
            int mid = left + (right - left) / 2;
            if (nums[mid] == target) {
                return mid;
            }
            
            // 先根据 nums[0] 与 target 的关系判断目标值是在左边的有序子数组还是右边的有序子数组
            if (target >= nums[0]) {
                // 目标值在左边有序数组中时，若 mid 在右边有序数组，则将 mid 索引的值改成 inf
                if (nums[mid] < nums[0]) {
                    nums[mid] = Integer.MAX_VALUE;
                }
            } else {
                // 目标值在右边有序数组中时，若 mid 在左边有序数组，则将 mid 索引的值改成 -inf
                if (nums[mid] >= nums[0]) {
                    nums[mid] = Integer.MIN_VALUE;
                }
            }

            if (nums[mid] < target) {//target在右边有序子数组，缩小目标范围为右半部分
                left = mid + 1;
            } else {//target在左边有序子数组，缩小目标范围为左半部分
                right = mid - 1;
            }
        }
        return -1;
    }
}