题目描述
假设按照升序排序的数组在预先未知的某个点上进行了旋转。
(例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2])。
请找出其中最小的元素。
你可以假设数组中不存在重复元素。
示例1
输入: [3,4,5,1,2] 输出: 1
示例2
输入: [4,5,6,7,0,1,2] 输出: 0
题解
这题如果直接遍历一遍的话,时间复杂度是 ,也能过。但是这题显然想要你更快,也就是用 的时间复杂度来做出来,那我们只能选择用二分法。
因为序列从中间切开来,然后调换过顺序,所以是先上升,再下降一下,然后再上升。并且第二段上升的最大值 是一定小于第一段上升的最小值 的,所以最小值一定是第二段的第一个数。
假设我们二分的时候,左端点 l ,右端点 r ,中间点是 m 。
如果 ,那说明左端点在第一段,右端点在第二段。这时如果 ,那么 m 也在第一段,所以 l 需要右移;否则的话 m 在第二段, r 需要左移。
如果 ,那么两个端点都在第二段,是单调上升的,那最小值一定就是 l 。
代码
c++
class Solution { public: int findMin(vector<int>& nums) { int l = 0, r = nums.size() - 1; while (l < r) { int m = (l + r) / 2; if (nums[l] > nums[r]) { if (nums[m] >= nums[l]) { l = m + 1; } else { r = m; } } else { r = l; } } return nums[r]; } };
python
class Solution: def findMin(self, nums: List[int]) -> int: l, r = 0, len(nums)-1 while l < r: m = (l + r) // 2 if nums[l] > nums[r]: if nums[m] >= nums[l]: l = m + 1 elif nums[m] < nums[r]: r = m else: r = l return nums[r]