[Leetcode]搜索旋转排序数组
题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [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
解题思路
- 时间复杂度是O(log n),就考虑使用二分,但是与二分查找不同的是,升序排序数组nums在某个位置进行了旋转,那么,如果中间位置为mid,左指针为left,右指针为right,分为三种情况:
- target=nums[mid]
- mid左边是升序序列
- target在 [left,mid) 里,right=mid-1
- target不在 [left,mid) 里,left=mid+1
- mid右边是升序序列
- target在 (mid,right] 里,left=mid+1
- target不在 (mid,right] 里, right=mid-1
实现代码
class Solution {
public:
int search(vector<int>& nums, int target) {
int len=nums.size();
if(len==0)
return -1;
int left=0,right=len-1;
while(left<=right){
int mid=(left+right)/2;
if(target==nums[mid])
return mid;
if(nums[left]<=nums[mid]){//如果左边是递增
if(target>=nums[left]&&target<=nums[mid])//如果target在左边
right=mid-1;
else
left=mid+1;
}
else{//如果右边是递增
if(target>=nums[mid]&&target<=nums[right])//如果target在右边
left=mid+1;
else
right=mid-1;
}
}
return -1;
}
};