您的位置:首页 > 其它

Leetcode 81. Search in Rotated Sorted Array II

2018-02-05 08:57 726 查看
原题:

Follow up for "Search in Rotated Sorted Array":

What if duplicates are allowed?

Would this affect the run-time complexity? How and why?

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
).

Write a function to determine if a given target is in the array.

The array may contain duplicates.

解决方法:
由于重复数据的存在,这道题的逻辑更加复杂一些,通过mid 和hi的比较,可以确定mid是在哪一个区间里面:

- 如果mid 大于hi,说明mid是在原来的递增区间,然后比较target和low与mid的关系,确定是移动lo还是hi;

- 如果mid 小于hi, 说明mid是在旋转过去的递增区间,然后比较target和hi与mid的关系,确定是移动lo还是hi;

- 如果mid等于hi,我们就只能将hi往前移动1位。

代码:
bool search(vector<int>& nums, int target) {
if (nums.empty())
return false;

int lo = 0, hi = nums.size() -1;
while(lo <= hi){
int mid = lo + ( (hi -lo) >> 1 );
if (nums[mid] == target)
return true;

if (nums[mid] > nums[hi]){
if (nums[mid] > target && nums[lo] <= target){
hi = mid - 1;
}else{
lo = mid + 1;
}
}else if (nums[mid] < nums[hi]){
if (target > nums[mid] && target <= nums[hi]){
lo = mid + 1;
}else{
hi = mid - 1;
}
}else{
--hi;
}
}

return nums[lo] == target;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  C Leetcode