题目大意
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。
搜索一个给定的目标值,如果数组中存在这个目标值,则返回它的索引,否则返回 -1 。
你可以假设数组中不存在重复的元素。
你的算法时间复杂度必须是 O(log n) 级别。
解题思路
题目要求复杂度为 O(log n) 级别,因此考虑二分法。
设置左右指针left和right,计算当前区域中点索引mid。
-
nums[mid] == target:返回mid即可;
-
nums[mid] >= nums[left]:证明mid左边是连续递增的部分([left,mid,right],假设在left~mid中有不是递增的部分,且没有重复元素,位置为k。则left ~ k-1部分递增,k-1 ~k衰减,k ~mid递增,由题目可知,left ~ k-1部分是有序数组尾部移动过来的,因此left ~ k-1部分的任意一个数字大于该数组其余部分的数字,与nums[mid] >= nums[left]矛盾。因此mid左边连续递增)。证明完左侧连续递增后,如果target在nums[left]和nums[mid]范围内,则表明targe如果在数组中,一定在左部分。反之在右部分。
-
nums[mid] < nums[left]:证明mid右边是连续递增的部分(证明同上)。此时如果target在nums[mid]和nums[right]范围内,则表明target如果在数组中,则一定是在mid右部分,反之在左部分。
class Solution {
public:
int search(vector<int>& nums, int target) {
if (nums.size() == 0)
return -1;
int left = 0, right = nums.size() - 1, mid = 0;
while(left < right){
mid = left + (right - left) / 2;
if (nums[mid] == target)
return mid;
//左部分是递增的
if (nums[mid] >= nums[left]){
if (target < nums[mid] && target >= nums[left])
right = mid - 1;
else
left = mid + 1;
}
//右部分是递增的
else{
if (target > nums[mid] && target <= nums[right])
left = mid + 1;
else
right = mid - 1;
}
}
return nums[left] == target ? left : -1;
}
};