Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7
might become 4 5 6 7 0 1 2
).
Find the minimum element.
You may assume no duplicate exists in the array.
这道题暴力算是O(n),所以肯定是要寻找小于O(n)的算法,注意到题目中array是sorted,指向了binary search。事实上array的小于O(n) 的算法只有binary search了把。
具体到这道题而言,如何二分查找呢?开始也没有什么头绪,不过试了几道例子之后就有一些感觉了。我们发现,二分查找的中间的值要不比两端的都高,这个时候最小值在[mi,hi]之中,要不然比两端的都低,这个时候最小值在[lo,mi]之中,要不然比最左边高比最右边低,这个时候最小值是最左边的。
这道题的思路最开始很不清晰,瞎想了很久,后来发现只要试几个例子马上找到一种简单的规律。所以以后看到没头绪的题先联想之前的题目,如果没有的话,试几个例子看看有没有简单做法,如果还没有再思考通用方法。
AC 代码:
class Solution {
public:
int findMin(vector<int>& nums) {
int lo = 0, hi = nums.size()-1;
while (hi-lo > 1) {
int mi = (lo + hi) / 2;
if (nums[mi] < nums[lo] && nums[mi] < nums[hi]){
hi = mi;
} else if (nums[mi] > nums[lo] && nums[mi] > nums[hi]){
lo = mi;
} else {
return nums[lo];
}
}
return min(nums[hi],nums[lo]);
}
};