题目
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).
Find the minimum element.
The array may contain duplicates.
思路
一个有序数组的开始一部分元素被搬到数组的末尾,是旋转有序数组,题目要求找到旋转数组的最小值。这道题的关键在于这个被旋转的有序数组中是有重复的元素。
最直接的解法就是从头到尾遍历一遍数组,找到最小的元素,这是一个O(n)的方法。
代码如下:
class Solution {
public:
int findMin(vector<int>& nums) {
int min = nums[0];
for(int i = 1;i < nums.size();i++)
{
if(nums[i] < min)
min = nums[i];
}
return min;
}
};
但这个思路并没有用到输入数组是一个旋转数组。对于这个旋转数组其实可以被分为两个有序的子数组,而两个子数组中间的数就是最小的数。所以可以类似于二分查找,分别有两个指针指向数组第一个元素和最后一个元素,按照题目的说法,第一个元素肯定大于最后一个元素。
找到数组最中间的数,跟最后一个元素做对比有三种情况:
1)如果最后一个元素小于最中间的数,则说明最中间的数在第一个递增子序列中,将第二个指针移到中间。
2)如果最后一个元素大于中间的数,则说明中间的数在第二个子序列中,则将第一个指针移动到中间。
3)如果最后一个元素等于中间的数,这时如果第一个元素也等于中间的数,则需要顺序查找,分别移动两个指针向中间靠拢。如果中间的数不等于第一个元素,则将第二个指针移到中间。
移动两个指针的时候,都会将查找范围缩小到原来的一半。
代码
class Solution {
public:
int findMin(vector<int>& nums) {
if(nums.size() == 1)
return nums[0];
int index1 = 0;
int index2 = nums.size()-1;
int mid = index1;
while(index1<index2)
{
mid = (index1+index2)/2;
if(nums[mid] > nums[index2])
index1 = mid+1;
else if(nums[mid] < nums[index2])
index2 = mid;
else
{
if(nums[index1] == nums[mid])
{
index1++;
index2--;
}
else
index2 = mid;
}
}
return nums[index2];
}
};