已知一个长度为 n 的数组,预先按照升序排列,经由 1 到 n 次 旋转 后,得到输入数组。例如,原数组 nums = [0,1,4,4,5,6,7] 在变化后可能得到:
若旋转 4 次,则可以得到 [4,5,6,7,0,1,4]
若旋转 7 次,则可以得到 [0,1,4,4,5,6,7]
注意,数组 [a[0], a[1], a[2], …, a[n-1]] 旋转一次 的结果为数组 [a[n-1], a[0], a[1], a[2], …, a[n-2]] 。
给你一个可能存在 重复 元素值的数组 nums ,它原来是一个升序排列的数组,并按上述情形进行了多次旋转。请你找出并返回数组中的 最小元素 。
示例 1:
输入:nums = [1,3,5]
输出:1
示例 2:
输入:nums = [2,2,2,0,1]
输出:0
提示:
n == nums.length
1 <= n <= 5000
-5000 <= nums[i] <= 5000
nums 原来是一个升序排序的数组,并进行了 1 至 n 次旋转
思路:二分法(来自leetcode官方题解)
首先,根据题意,所谓旋转数组就是把一个有重复数字的有序数组,将末尾的一部分整体的移动到头部。
所以寻找最小元素的问题就可以转换为寻找两个有序数组的交界点。如图所示:
定义二分做边界为low,右边界为high,二分中间点为pivot。
一:首先考虑没有重复元素的时候,这种情况即[leetcode]153.寻找旋转排序数组中的最小值的情况。
则此时有以下两种情况:
情况一:nums[pivot]<nums[high]
。如图所示:
这种情况之下,在分界点之后的部分一定是递增有序的,且分界点(最小值)要么在pivot之前,要么就是pivot自己,于是区间调整为right=pivot
。
情况二:nums[pivot]>nums[high]
。如图所示:
这种情况下数组的前半部分一定是顺序递增的,且最小值(分界点)一定在pivot之后,所以需要令left=pivot+1
,这里之所以加一是因为最小值点一定不可能是nums[piovt],(起码至少比nums[high]要大)。
当左右端点重合的时候即为答案。
AC代码:(C++)
class Solution {
public:
int findMin(vector<int>& nums) {
int left = 0, right = nums.size() - 1;
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[mid] < nums[right]) {
//情况1
right = mid;
} else {
//情况2
left = mid + 1;
}
}
return nums[left];
}
};
二:当加入重复元素的时候,也就是本题的情况。
情况三:有重复元素的情况下,除了之前的情况,还会出现一种特殊情况,即nums[pivot]==nums[high]
。如图所示:
在这种情况下,无法判断分界点应该在左半部分区间还是在右半部分区间,因此不能随便的忽略某一部分元素。但是由于nums[pivot]与nums[high]相等,所以无论nums[high]到底是不是最小值,都有一个nums[pivot]可以替代,于是我们可以舍弃掉nums[high]这个点,令high向左边移动一位,重新进行二分的过程。极端情况下,所有元素都相同,high需要逐个移动到left,此时时间复杂度为O(n)。
于是本题的代码只需要在上题的基础上增加对情况3的一个特殊处理。
AC代码:(C++)
class Solution {
public:
int findMin(vector<int>& nums) {
int left = 0, right = nums.size() - 1;
while (left < right) {
int mid = left + (right - left) / 2;
if (nums[mid] < nums[right]) {
//情况1
right = mid;
} else if (nums[mid] > nums[right]) {
//情况2
left = mid + 1;
} else
//情况3
right--;
}
}
return nums[left];
}
};