您的位置:首页 > 其它

[leetcode] 33. Search in Rotated Sorted Array

2016-01-22 19:18 309 查看
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;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  leetcode Binary Search