leetcode81. 搜索旋转排序数组 II

题目

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,0,1,2,2,5,6] 可能变为 [2,5,6,0,0,1,2] )。

编写一个函数来判断给定的目标值是否存在于数组中。若存在返回 true，否则返回 false。

示例 1:

输入: nums = [2,5,6,0,0,1,2], target = 0
输出: true
示例 2:

输入: nums = [2,5,6,0,0,1,2], target = 3
输出: false

解题思路

使用二分查找，先判断nums[mid]在旋转点的左侧或者右侧，即 nums[mid] <= nums[high] 则在右侧；nums[mid] >= nums[low] 则在左侧。
在右侧时，如果 target > nums[mid] && target <= nums[high] target只可能在区间[mid+1, high]中；否则，target只能在区间[low,mid-1]中。
左侧也同理，如果 target < nums[mid] && target >= nums[low] target只可能在区间[low, mid-1]中；否则只能在[mid+1, high]中。
最后的一点就是在判断nums[mid]在旋转点哪一侧时，如果有重复的数字，会导致判断错误，例如nums={1,1,3,1}，target = 3； 无法判断nums[mid]在哪一侧。

代码如下

class Solution {
    public boolean search(int[] nums, int target) {
        if(nums == null || nums.length < 1) return false;
		int low = 0, high = nums.length-1;
		while(low <= high) {
			//去掉重复的点，不然在下面的判断mid在旋转点左右侧会有影响 如{1,1,3,1}则无法判断mid的位置
			while(low+1 <= high && nums[low+1] == nums[low]) low++;
			while(high-1 >= low && nums[high-1] == nums[high]) high--;
			int mid = (low + high) >> 1;
			if(target == nums[mid]){
				return true;
			} else if(nums[mid] <= nums[high]) {
				//mid在旋转点右边
				if(target > nums[mid] && target <= nums[high]) {
					low = mid + 1;
				} else {
					high = mid - 1;
				}
			} else if(nums[mid] >= nums[low]) {
				//mid在旋转点左边
				if(target < nums[mid] && target >= nums[low]) {
					high = mid - 1;
				} else {
					low = mid + 1;
				}
			}
		}
		return false;
    }
}

提交结果

成功
显示详情
执行用时 : 2 ms, 在Search in Rotated Sorted Array II的Java提交中击败了91.00% 的用户
内存消耗 : 37.4 MB, 在Search in Rotated Sorted Array II的Java提交中击败了52.09% 的用户