Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7
might become 4 5 6 7 0 1 2
).
You are given a target value to search. If found in the array return its index, otherwise return -1.
You may assume no duplicate exists in the array.
在这样一个数组中找到target,这个数组是递增序列的变体,如4 5 6 7 0 1 2,或5 6 7 0 1 2 3 4
解决方法,可以采用二分法思想:
------------------------------------------------------------------------------------------------------------------------------
假设数组是A,取左边缘为l,右边缘为r,还有中间位置是m。在每次迭代中,分三种情况:
(1)如果target==A[m],那么m就是我们要的结果,直接返回m;
(2)如果A[m]<A[r](情况就类似于:如5 6 7 0 1 2 3 4中,m = 3,r = 7。A[3] = 0 < A[7] = 1)
那么说明从m到r右半部分一定是有序的,此时只需要判断target是不是在m到r之间,如果是则把左边缘移到m+1,否则就target在另一半,即把右边缘移到m-1。
(3)如果A[m]>=A[r](情况就类似于:如4 5 6 7 0 1 2中,m = 3,r = 7。A[3] = 7 >= A[7] = 0)
那么说明从l到m左半部分一定是有序的,同样只需要判断target是否在这个范围内,相应的移动边缘即可
-------------------------------------------------------------------------------------------------------------------------------
根据以上方法,每次可以切掉一半的数据,所以算法的时间复杂度是O(logn),空间复杂度是O(1)。代码如下:
class Solution {
public:
int search(vector<int>& nums, int target){
if(nums.capacity() == 0)
return -1;
int l = 0; //左边缘
int r = nums.capacity() - 1; //右边缘
while(l <= r){
int m = (l + r) / 2;
if(target == nums[m])
return m;
if(nums[m] < nums[r]){ //情况类似5 6 7 0 1 2 3 4(从中间到最后有序)
if(target > nums[m] && target <= nums[r]) //这个数在右半部分
l = m + 1;
else
r = m - 1;
}
else{ //情况类似4 5 6 7 0 1 2(从左边缘到中间有序)
if(target >= nums[l] && target < nums[m]) //这个数在左半部分
r = m - 1;
else
l = m + 1;
}
}
return -1;
}
};