剑指 11 旋转数组的最小数字
原题目
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个递增排序的数组的一个旋转,输出旋转数组的最小元素。例如,数组 [3,4,5,1,2] 为 [1,2,3,4,5] 的一个旋转,该数组的最小值为1
示例 1:
输入:[3,4,5,1,2]
输出:1
示例 2:
输入:[2,2,2,0,1]
输出:0
示例 3:
输入:[1,3,5]
输出:1
示例 4:
输入:[1,1,1]
输出:1
示例 5:
输入:[1,1,1,0,1]
输出:0
考查知识点
查找–>二分查找
好的解法
输入的数组中明显包含两个已经排好序的子数组,第一个子数组中所有数字都大于等于第二个子数组中的所有数字。对于排好序的数组,最容易想到的就是二分查找。
二分查找需要确定三个关键点:起点
、中点
、终点
。二分查找就是在不断将起点
与终点
更新到中点
的位置上,不断缩小查找范围,直到这三个点其中的一个点指到了目的元素上。
对于已经排序好的一个数组,常用的操作是将起点
与终点
指在数组的两端,即指向最小数与最大数。但是本题包含两个排序的数组,因此不能保证原数组两端就是最小值和最大值,没关系,我们先按照最传统的指法来,即同样是让起点与终点分别指在数组的两端。
确定好起点与终点之后,现在看中点
,自然是指在了原数组中间,之前一直特别纠结整个数组长度是奇数还是偶数对中点的影响,只要算法(即对起点与终点的更新
)对了,数组长度的奇偶是没有影响的。中点有两种情况,(1)中点元素 >= 起点元素,此时中点元素在第一个子数组中;(2)中点元素 <= 终点元素,此时中点元素在第二个子数组中。目标
是找到最小的元素,那么可以确定的是,最小的元素一定是第二个子数组中的第一个元素。
下面分别考虑起点与终点的更新:
(1)中点元素 >= 起点元素,此时中点元素在第一个子数组中,目标肯定在中点右边,起点<–中点
(2)中点元素 <= 终点元素,此时中点元素在第二个子数组中,目标肯定在中点左边或者就是中点, 终点<–中点
class Solution {
public:
int minArray(vector<int>& numbers) {
if (numbers.empty())
return -1;
int p1 = 0;
int p2 = numbers.size() - 1;
int p_mid = p1;
while (numbers[p1] >= numbers[p2])
{
if (p2-p1 == 1) // p2与p1相邻,p2所指就是小数组中第一个数字
{
p_mid = p2;
break;
}
p_mid = (p1 + p2) / 2;
if (numbers[p_mid] >= numbers[p1]) //加“=”应对示例2
p1 = p_mid;
else if (numbers[p_mid] <= numbers[p2])
p2 = p_mid;
}
return numbers[p_mid];
}
};
上面代码基本包括了一般情况,比如示例1,但是有几种特殊情况是在提交时才发现错误
示例2:num[p1]=2, num[p_mid]=2, 此时if语句如果不加等于号,会陷入死循环
示例3:这种整体是一个排序数组的情况可以看作是旋转了0个数字的数组,当然也要算边界条件,现在明显最小的元素就在p1的位置,这也是将p_mid初始化为p1的原因
示例4与示例5:对于这种情况,起点中点终点指向的元素数值相同,此时上面代码的判断就失效了,因为不能正确的更新起点与终点,这时就要采取顺序查找,找到数组中的最小数字
完整代码如下:
class Solution {
public:
int minArray(vector<int>& numbers) {
if (numbers.empty())
return -1;
int p1 = 0;
int p2 = numbers.size() - 1;
int p_mid = p1;
while (numbers[p1] >= numbers[p2])
{
if (p2-p1 == 1) // p2与p1相邻,p_mid所指就是小数组中第一个数字
{
p_mid = p2;
break;
}
p_mid = (p1 + p2) / 2;
if (numbers[p1]==numbers[p2] && numbers[p_mid]==numbers[p1])
return min_in_order(numbers, p1, p2);
if (numbers[p_mid] >= numbers[p1])
p1 = p_mid;
else if (numbers[p_mid] <= numbers[p2])
p2 = p_mid;
}
return numbers[p_mid];
}
int min_in_order(vector<int> nums, int p1, int p2)
{
int min = nums[p1];
for (int i=p1+1; i<p2; ++i)
{
if (nums[i] < min)
min = nums[i];
}
return min;
}
};