Leetcode Find Minimum in Rotated Sorted Array 题解

时间:2023-03-09 20:35:25
Leetcode Find Minimum in Rotated Sorted Array 题解

Leetcode Find Minimum in Rotated Sorted Array

题目大意:

对一个有序数组翻转, 就是随机取前K个数,移动到数组的后面,然后让你找出最小的那个数。注意,K有可能是0,也就是没有翻转。

毫无疑问,遍历一次肯定能够找到,但这样时间复杂度是O(n)。假设你在面试的时候遇到这种问题,你这样回答面试官肯定不会惬意的。我们接下来讨论有没有什么更快的方法。O(nlogn)??

我还是把O(N)的代码贴出来,不知道为什么leetcode上竟然不超时。

//O(n)
class Solution {
public:
int findMin(vector<int> &num) {
int minval = 0x3f3f3f3f;
int len = num.size();
for (int i = 0; i < len; i++) {
minval = min(minval, num[i]);
}
return minval;
}
};

既然数字開始是有序的,我们第一个应该想到的方法就是二分。其实就是能够二分。 假设进行的翻转,肯定会变成两部分有序数组。第一部分的不论什么一个数都大于第二部分的,第二部分中最后一个数肯定是最大的。这样二分的推断条件就是,仅仅要mid位置的值大于最后一个数。肯定能确定mid在第一部分中,然后往右二分就可以,反之往左。

我本人写了两次代码,第一份非常不尽人意,首先对没有翻转的情况做了特判。然后一直遇到二分边界的问题。仅仅要二分两个数的时候就会死循环,索性就加了特判,两个数的时候直接输出最小的一个,代码冗长混乱。思路也不严谨,先贴出来做个反例。

//O(nlogn) bad
class Solution {
public:
int findMin(vector<int> &num) {
int len = num.size();
if (num[0] <= num[len-1])
return num[0];
int l = 0, r = len-1;
while (l < r) {
int mid = (l+r)>>1;
if (num[mid] > num[l]) {
if (num[mid] > num[0])
l = mid+1;
if (num[mid] < num[len-1])
r = mid;
}
else if (num[mid] < num[r])
r = mid;
else
return min(num[l], num[r]);
}
return num[l];
}
};

接下来就是改动后的精简代码,事实上不须要特判,又一次写了二分的推断条件,然后代码瞬间短了好多,也易于理解。

//O(nlogn) good
class Solution {
public:
int findMin(vector<int> &num) {
int len = num.size();
int l = 0, r = len-1;
while (l < r) {
int mid = (l+r)>>1;
if (num[mid] > num[r])
l = mid+1;
else
r = mid;
}
return num[l];
}
};