Description
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个递增排序的数组的一个旋转,输出旋转数组的最小元素。例如,数组 [3,4,5,1,2] 为 [1,2,3,4,5] 的一个旋转,该数组的最小值为1。
示例 1:
输入:[3,4,5,1,2]
输出:1
示例 2:
输入:[2,2,2,0,1]
输出:0
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/xuan-zhuan-shu-zu-de-zui-xiao-shu-zi-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
Solution
可以使用减治,这里排除中位数选择的条件是nums[mid] > nums[r]
,另外还需要加一个缩小右边界的情况。
class Solution:
def findMin(self, nums: List[int]) -> int:
l, r = 0, len(nums)-1
while l < r:
mid = (l+r) >> 1
if nums[mid] > nums[r]: # 排除中位数
l = mid + 1
elif nums[mid] == nums[r]:
r -= 1
else:
r = mid
print(l,mid,r)
return nums[l]