题目链接:https://leetcode.cn/problems/search-in-rotated-sorted-array/
参考链接:https://leetcode.cn/problems/search-in-rotated-sorted-array/solution/duo-si-lu-wan-quan-gong-lue-bi-xu-miao-dong-by-swe/
题目描述
示例
提示
思路分析
首先,本题很直观让人想到二分查找,但是与传统的二分查找的不同点在于,数组进行了翻转,原来有序的升序数组被分成了两段有序。
对于所找的目标target
,我们可以比较targte
和nums[0]
的大小来判断target
可能被分在了左段还是右段。代码如下:
if(nums[0]<=target){
//target一定在数组的左侧
}else{
//反之,target在数组的右侧
}
对于旋转数组 nums = [4,5,6,7,0,1,2]
,例如 target = 5
, 目标值在左半段,因此在[4, 5, 6, 7, inf, inf, inf]
这个有序数组里找就行了;例如 target = 1
, 目标值在右半段,因此在 [-inf, -inf, -inf, -inf, 0, 1, 2]
这个有序数组里找就行了。
然后,判断两侧数组的有序性:
int mid = (right-left)/2+left;
if(nums[mid]==targte)return mid;
if(nums[0]<=target){
//target一定在数组的左侧
if(nums[mid]<nums[0]){
//如果中点的元素小于最左侧的元素,很明显右侧是有序的
//对于target来说,他在数组的左侧,但是右侧是有序的,所以我们需要修改最右侧的位置,用max值记录
nums[mid]=Integer.MAX_VALUE;
}
}else{
//反之,target在数组的右侧
if(nums[mid]>=nums[0]){
//左侧有序
//同理,target在数组的右侧,数组左侧有序,用min值记录
nums[mid]=Integer.MIN_VALUE;
}
}
//最后,修改左右两侧的位置
if(nums[mid]<target){
//如果说中点的值小于target
left = mid;
}else{
//如果说终点的值大于等于target
right = mid;
}
最后返回的是right
,因为right
记录了大于等于target
的值,这个思想参考bilibili
视频:https://www.bilibili.com/video/BV1d54y1q7k7/
完整代码
class Solution {
public int search(int[] nums, int target) {
//找到nums的有序部分,并且target属于这部分
int m = lower_bound(nums, target);
return m;
}
public int lower_bound(int[] nums, int target){
int left = -1;
int right = nums.length;
while(left + 1!=right){
int mid = (right-left)/2 + left;
if(nums[mid]==target)return mid;
//根据target与nums[0]的关系判断是在左半段还是右半段
if(nums[0]<=target){
//目标在左半段
if(nums[mid]<nums[0]){
//顺序在右半段
nums[mid] = Integer.MAX_VALUE;
}
}else{
//目标在右半段,顺序在左半段
if (nums[mid]>=nums[0]) {
nums[mid] = Integer.MIN_VALUE;
}
}
if(nums[mid]<target){
left = mid;
}else{
right = mid;
}
}
// System.out.println(right);
return right==nums.length||nums[right]!=target?-1:right;
}
}