【题目】
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。
输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。
例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。
NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
【思路】
如果直接使用顺序查找,时间复杂度为 O ( n ) O(n) O(n),完全没用到旋转数组的特性,显然不合适。
观察可以发现,旋转后的数组分成两个序列,而两个序列都是非递减排序,前面的序列的元素都大于或者等于后面序列的元素。另外最小的元素刚好是这两个序列的右边分界线。旋转后的数组在一定程度上也是排序的,可以采用二分查找,时间复杂度为 O ( l o g 2 n ) O(log_2{n}) O(log2n)。
用二分查找的思想,设置左指针left,右指针right和中间指针mid。
当mid指向的值大于或等于left指向的值,说明要找的最小元素在mid后面,所以令left等于mid。
当mid指向的值小于或等于right指向的值,说明要找的最小元素在mid前面,所以令right等于mid。
当left和right相邻时,可以判断出最小元素就是right指向的值。
特殊情况是,旋转的元素数目为0,即不旋转,那么第一个元素就是最小元素。
【代码】
- 改进前(牛客网提交也能通过)
class Solution {
public:
int minNumberInRotateArray(vector<int> rotateArray) {
int size = rotateArray.size();
if (size == 0) {
return 0;
}
int left = 0;
int right = size - 1;
int mid = 0;
while (rotateArray[left] >= rotateArray[right]) { // 有旋转
if (right - left == 1) {
return rotateArray[right];
}
mid = (left + right) / 2;
if (rotateArray[mid] >= rotateArray[left]) {
left = mid;
}
else{
right = mid;
}
}
return rotateArray[0]; // 无旋转,第1个为最小值
}
};
- 改进后(完整版)
看了其他人的一些答案,发现还有一种特殊情况没考虑。假设初始数组是[1, 2, 2, 2, 2],旋转后是[2, 2, 2, 1, 2],计算left、right和mid时发现这三个位置的数都是2,此时上面的代码无法判断最小值1是在左边还是右边,所以需要特判并进行顺序查找。
class Solution {
public:
int minNumberInRotateArray(vector<int> rotateArray) {
int size = rotateArray.size();
if (size == 0) {
return 0;
}
int left = 0;
int right = size - 1;
int mid = 0;
while (rotateArray[left] >= rotateArray[right]) { // 有旋转
if (right - left == 1) {
return rotateArray[right];
}
mid = (left + right) / 2;
// 特判,当left、right和mid位置的数相同
if (rotateArray[left] == rotateArray[mid] &&
rotateArray[mid] == rotateArray[right]) {
int result = rotateArray[left];
for (int i = left + 1; i <= right; i++) {
if (rotateArray[i] < result) {
result = rotateArray[i];
}
}
return result;
}
if (rotateArray[mid] >= rotateArray[left]) {
left = mid;
}
else{
right = mid;
}
}
return rotateArray[0]; // 无旋转,第1个为最小值
}
};