题目:
Follow up for "Find Minimum in Rotated Sorted Array":
What if duplicates are allowed?Would this affect the run-time complexity? How and why?
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
).
Find the minimum element.
The array may contain duplicates.
假设一个有序数组在数组中某个未知的点处被旋转了,找出旋转后数组中最小的元素。(数组中可能含有重复的数)。
思路:
请先参看:查找旋转有序数组中的最小值。本题与上一题的差别在于,上一题中数组的前半段数字一定比后半段数字大,第一个数字一定比后半段所有数字大。而本题中,由于可以有重复数字,所以前半段未必一定大于后半段,有可能有些数字相等。 比如1 1 1 0 1,或者1 0 1 1 1 。这个时候无法确定最小值在前半段还是后半段,所以不能把 mid 或者 mid+1 直接赋给 pre 或者 next ,而是让 pre++,一直往后。如果碰到 num [ pre ] < num [ next ] ,则说明当前值 num [ pre ] 一定是最小值。
代码:
class Solution {
public:
int findMin(vector<int> &num)
{
int pre = 0;
int next = num.size() - 1;
if(num.empty())
return 0;
//循环结束条件是pre == next 或者 num[pre] < num[next],后者是因为
while(pre < next && num[pre] >= num[next])
{
int mid = (pre + next)/2;
if(num[mid] > num[next])
pre = mid + 1;
else
if(num[mid] < num[next])
next = mid;
//mid与next对应的值相等,比如1 1 1 0 1,或者1 0 1 1 1,无法判断在前半段还是后半段
else
pre ++;
}
return num[pre];
}
};