leetcode 33. 搜索旋转排序数组

2019-04-21  本文已影响0人  topshi

题目描述

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

思路:
有序数组的二分查找时,定义两个指针lr


而我们这题的数组有可能发生了旋转,那么就不能单纯依据targetnums[mid]的比较来判断targetmid的左还是在右。

class Solution {
    public int search(int[] nums, int target) {
        if(nums.length == 0) return -1;
        int l = 0, r = nums.length - 1;
        while(l <= r){
            int mid = l + ((r - l) >> 1);
            if(nums[mid] == target){
                return mid;
            }
            //左部分无序,右部分有序
            if(nums[l] > nums[mid]){
                if(target >= nums[mid] && target <= nums[r]){
                    l = mid + 1;
                }else{
                    r = mid - 1;
                }
                //右部分无序,左部分有序
            }else if(nums[mid] > nums[r]){
                if(target <= nums[mid] && target >= nums[l]){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
                //左右部分都有序
            }else{
                if(target <= nums[mid] && target >= nums[l]){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
            }
        }
        return -1;
    }
}

这里的mid也纳入了左或右部分,而不是左:l->mid-1右:mid+1->r。这是为了减少边界的处理。
以下代码:左:l->mid-1右:mid+1->r

class Solution {
    public int search(int[] nums, int target) {
        if(nums.length == 0) return -1;
        int l = 0, r = nums.length - 1;
        while(l <= r){
            int mid = l + ((r - l) >> 1);
            if(nums[mid] == target){
                return mid;
            }
            //左部分无序,右部分有序
            if(mid - 1 >= l && nums[l] > nums[mid - 1]){
                if(mid + 1 <= r && target >= nums[mid + 1] && target <= nums[r]){
                    l = mid + 1;
                }else{
                    r = mid - 1;
                }
                //右部分无序,左部分有序
            }else if(mid + 1 <= r && nums[mid + 1] > nums[r]){
                if(mid - 1 >= l && target <= nums[mid - 1] && target >= nums[l]){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
                //左右部分都有序
            }else{
                if(mid - 1 >= l && target <= nums[mid - 1] && target >= nums[l]){
                    r = mid - 1;
                }else{
                    l = mid + 1;
                }
            }
        }
        return -1;
    }
}

由于判断条件使用到mid - 1mid + 1的操作,可能会导致数组越界。

上一篇下一篇

猜你喜欢

热点阅读