leetcode原题链接:搜索旋转排序数组
上一篇:HOT65-在排序数组中查找元素的第一个和最后一个位置
题目描述
整数数组 nums
按升序排列,数组中的值 互不相同 。
在传递给函数之前,nums
在预先未知的某个下标 k
(0 <= k < nums.length
)上进行了 旋转,使数组变为 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]]
(下标 从 0 开始 计数)。例如, [0,1,2,4,5,6,7]
在下标 3
处经旋转后可能变为 [4,5,6,7,0,1,2]
。给你 旋转后 的数组 nums
和一个整数 target
,如果 nums
中存在这个目标值 target
,则返回它的下标,否则返回 -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
示例 3:
输入:nums = [1], target = 0 输出:-1
提示:
1 <= nums.length <= 5000
-104 <= nums[i] <= 104
nums
中的每个值都 独一无二- 题目数据保证
nums
在预先未知的某个下标上进行了旋转 -104 <= target <= 104
解题方法:二分查找。首先,对于一个mid分割的两个区间[left, mid)和(mid, right],根据旋转排序数组的定义,这两个区间至少有一个区间是有序的。判断[left, mid), (mid, right]两个区间,到底哪个是有序的空间,如果target在有序空间里,则对该区间采用二分查找。所以本题的关键在于,需要判断有序区间在哪里。
C++代码
#include <iostream>
#include <vector>
/*
* 1 2 3 4 5 6 7
* 4 5 6 7 1 2 3
*/
class Solution {
public:
int search(std::vector<int>& nums, int target) {
int n = nums.size();
int left = 0;
int right = n - 1;
while (left <= right) {
int mid = (left + right) / 2;
if (target == nums[mid]) {
return mid;
}
// mid分割的两个区间: [left,mid), (mid,right]必然有一个是有序的空间
if (nums[left] <= nums[mid]) { //[left, mid]是有序,注意这里是 <=,不能写成<,否则如果数组元素个数为1,会进入死循环
if (target >= nums[left] && target < nums[mid]) { //target位于区间[left, mid)
right = mid - 1;
} else {
left = mid + 1;
}
} else {//(mid, right]有序
if (target > nums[mid] && target <= nums[right]) { // target位于区间(mid, right]
left = mid + 1;
} else {
right = mid - 1;
}
}
}
return -1;
}
};