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];
}
};