Suppose a sorted array 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.
这道题是在前后两段颠倒后的有序数组中查找给定值,题目难度为Hard。
题目可以参考二分查找法,只是这里数组被颠倒过了,所以对中间元素的比对更加复杂。这里分别用l、m、r表示二分查找过程中左端、中间和右端的元素下标。
-
如果target==nums[m],则查找结束。
-
如果target>nums[m],此时分为三种情况。
-
如果target>nums[l],需比较nums[m]和nums[l]:若nums[m]>=nums[l],表明目标值在后半段,所以将l置为m+1;若nums[m]<nums[l],表明目标值在前半段,所以将r置为m-1。
-
如果target<nums[l],此时只有一种情况,目标值在后半段,将l置为m+1。
-
如果target==nums[l],则查找结束。
-
-
如果target<nums[m],此时也分为三种情况。
-
如果target<nums[l],需比较nums[m]和nums[l]:若nums[m]>=nums[l],表明目标值在后半段,所以将l置为m+1;若nums[m]<nums[l],表明目标值在前半段,所以将r置为m-1。
-
如果target>nums[l],此时只有一种情况,目标值在前半段,将r置为m-1。
-
如果target==nums[l],则查找结束。
-
-
如果循环结束没有找到,则返回-1。
整个判断过程还是比较绕的,也不一定用nums[l]、nums[m]和target三者进行比较,整个判断过程其实就是将三者依大小进行排列组合,根据各种情况进行二分查找。数组如果未颠倒也是可以查找的。具体代码:
class Solution {
public:
int search(vector<int>& nums, int target) {
int l = 0, r = nums.size()-1, m;
while(l <= r) {
m = (l + r) / 2;
if(target == nums[m])
return m;
else if(target > nums[m]) {
if(target > nums[l]) {
if(nums[m] >= nums[l]) l = m + 1;
else r = m - 1;
}
else if(target < nums[l]) {
l = m + 1;
}
else return l;
}
else {
if(target < nums[l]) {
if(nums[m] >= nums[l]) l = m + 1;
else r = m -1;
}
else if(target > nums[l]) {
r = m - 1;
}
else return l;
}
}
return -1;
}
};