LeetCode81——Search in Rotated Sorted Array II

LeetCode81——Search in Rotated Sorted Array II

加入重复数据的Search in Rotated Sorted Array

原题

Follow up for “Search in Rotated Sorted Array”:

What if duplicates are allowed?

在Search in Rotated Sorted Array题中，如果允许重复元素情况求解。

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.

无重复元素

进行二分查找，以中间元素

nums[mid]

为中心分为左右两边，必然出现下述情况的某一种

左边有序

右边有序

以左边有序为例，假如左边有序，那么我们判断

target

是否在左边子序列中，如果在，则对子序列进行二分搜索，否则对右边子序列做二分搜索。

代码(无重复元素Roated Sorted Array的二分查找)：

class Solution {
public:
int search(vector<int>& nums, int target) {
int nSize = nums.size();
int low = 0;
int high = nSize - 1;
while (low <= high)
{
int mid = low + (high - low) / 2;//防止溢出
if (target == nums[mid])//查找结果成功
return mid;
//这里的细节要用>=原因是计算mid的时候是向下取整有可能
//mid==low那么nums[mid]==nums[low]
//排除测试用例 [3,1] 1
else if (nums[mid] >= nums[low])//左边有序
{
if (target < nums[mid] && target>=nums[low])//target在左边
high = mid - 1;
else//target在右边
low = mid + 1;
}
else//右边有序
{
if (target > nums[mid] && target <= nums[high])
low = mid + 1;
else
high = mid - 1;
}
}
return -1;
}
};

有重复元素

那么现在问题来了，如何去除重复元素呢？

其实考虑的的还是一样，只不过比较的时候考虑一点，要考虑当nums[mid]==nums[low]时的情况，也就是如果给定序列为[6,6,6,6,6,4,5]，而查找的target是5时，我们在第一次循环中，得到的nums[mid]==nums[low]这个时候我们要舍去这个nums[low]，使low++，可以理解为整体右移（由于mid是向下取整，所以最坏情况下需要至少右移两次），这样就去掉重复部分了。

代码：

class Solution {
public:
bool search(vector<int>& nums, int target) {
int nSize = nums.size();
int low = 0;
int high = nSize - 1;
while (low <= high)
{
int mid = low + (high - low) / 2;
if (nums[mid] == target)
return true;
else if (nums[low] < nums[mid])//右边有序
{
if (target >= nums[low] && target<nums[mid])
high = mid - 1;
else
low = mid + 1;
}
else if (nums[low] > nums[mid])//左边有序
{
if (target <= nums[high] && target > nums[mid])
low = mid + 1;
else
high = mid - 1;
}
else//相等要"整体右移"
{
low++;
}
}
return false;
}
};

参考：

http://blog.csdn.net/linhuanmars/article/details/20588511