【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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法

方法1

先使用二分法找到数组在哪里分隔的，然后通过判断target在哪个区间，然后再次使用一次二分法。时间复杂度为O(N*LogN)，空间复杂度为O(1)

class Solution {
    public int search(int[] nums, int target) {
        int left  = 1;
        int right = nums.length - 1;
        int index = 0;
        while(left <= right){
            int mid = left + (right - left) / 2;
            if(nums[mid] < nums[mid-1]){
                index = mid;
                break;
            }else{
                if(nums[mid] > nums[0]){
                    left = mid + 1;
                }else{
                    right = mid - 1;
                }
            }
        }
        System.out.println(index);
        if(index == 0){
            return binarySearch(nums, 0, nums.length - 1, target);
        }

        if(target >= nums[0]){
            return binarySearch(nums, 0, index - 1, target);
        }else{
            return binarySearch(nums, index, nums.length-1, target);
        }

    }

    public int binarySearch(int[] nums, int left, int right, int target){
        while(left <= right && left >=0 && right <= nums.length){
            int mid = left + (right - left) / 2;
            if(nums[mid] == target){
                return mid;
            }else if(nums[mid] > target){
                right = mid -1;            
            }else{
                left = mid + 1;
            }
        }
        return -1;
    }
}

方法2

使用1次二分法，通过判断target和nums[mid]、target和nums[0]以及nums[0]和nums[mid]的大小关系，确定下一次的区间。推导结果如下图所示。时间复杂度为O(N*LogN)，空间复杂度为O(1)

class Solution {
    public int search(int[] nums, int target) {
        int left  = 0;
        int right = nums.length - 1;
        while(left <= right){
            int mid = left + (right - left) / 2;
            if(nums[mid] == target){
                return mid;
            }else if(nums[0] == target){
                return 0;
            }else{
                int sum = 0;
                if(nums[mid] < target){
                    sum++;
                }

                if(target < nums[0]){
                    sum++;
                }

                if(nums[mid] < nums[0]){
                    sum++;
                }

                if(sum % 2 == 1){
                    left = mid + 1;
                }else{
                    right = mid -1;
                }
            }
           
        }
        return -1;
    }
}