【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
解题思路
一个有序数组经过旋转操作后,可以看成两部分,而这两部分又分别是有序的,我们先找到tatget
落在数组的哪一半,再在那一半继续进行二分查找。
Java 实现
public int search (int[] nums, int target) {
int n = nums.length;
int lo = 0, hi = n - 1;
while (lo <= hi) {
int mid = (lo + hi) / 2;
if (nums[mid] == target)
return mid;
if (nums[mid] >= nums[lo]) {
if (target >= nums[lo] && target < nums[mid])
hi = mid - 1;
else
lo = mid + 1;
}
else if (nums[mid] <= nums[hi]) {
if (target > nums[mid] && target <= nums[hi])
lo = mid + 1;
else
hi = mid - 1;
}
}
return -1;
}
心得体会
二分查找的变种,要注意哪些地方取等哪些地方不取等。
在判断target
和nums[mid]
不能取等,因为它们正好相等的时候要返回mid
,所以其他地方的判断不能取等。