描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
首先如果这个数组是直接完全有序的数组,那么简单的二分查找就能解决问题,但是这里是在某个未知地方进行了旋转的,所以这里要将查找范围左移或右移的条件发生了一些改变。
至于这个新条件如何确定,我们要先明确一件事,那就是旋转位置的左边的数都大于或等于nums[0],旋转位置右边的数都小于nums[0],并且旋转位置两边的两个子列都各自有序。
现在来考虑什么时候把查找范围右移(即代码中的low=mid+1):
目前的mid指向的位置和target所在的位置(如果数组中存在target)在同一个有序子列且nums[mid]<target,即nums[mid]<target<nums[0]或者nums[0]<=nums[mid]<target。这时其实和有序数组的二分查找是一样的,如果nums[mid]<target说明目标target在mid的右侧,向右查找;
mid指向的位置在旋转位置左边而target在旋转位置右边(如果数组中存在target),即target < nums[0]<=nums[mid]。
对于其他情况,如果nums[mid] == target那就直接返回mid,再有其他情况那自然是向左查找了, 即high=mid-1。
实现
public int search(int[] nums, int target) {
if (nums.length == 0) return -1;
int low = 0, high = nums.length - 1, mid;
while (low <= high) {
mid = (high + low) / 2;
if (nums[mid] == target)
return mid;
if ((nums[mid] < target && target < nums[0]) || (nums[mid] < target && nums[mid] >= nums[0])
|| (nums[mid] >= nums[0] && target < nums[0]))
low = mid + 1;
else
high = mid - 1;
}
return -1;
}