154. 寻找旋转排序数组中的最小值 II（python3）

题目：https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array-ii/

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

请找出其中最小的元素。

注意数组中可能存在重复的元素。

示例 1：

输入: [1,3,5]
输出: 1
示例 2：

输入: [2,2,2,0,1]
输出: 0
说明：

这道题是 寻找旋转排序数组中的最小值 的延伸题目。
允许重复会影响算法的时间复杂度吗？会如何影响，为什么？

 

解题思路：

二分查找，每次二分点两边一定有一边是有序的（或值全相同），用key记录最小数，每次用key与有序边的最小值相比，最终得到结果。
此题需注意存在重复元素，每次比较nums[mid]与nums[start]或nums[end]，如相等则start++或end--舍去相同值。key需在此处与nums[mid]对比，如nums[mid]小则更新，否则会出现[1,3,5]，当mid、start、end都指向下标0时key没被记录就跳出循环的情况。

 

代码：

import sys

class Solution:
    def findMin(self, nums: List[int]) -> int:
        start = 0
        end = len(nums) - 1
        
        key = sys.maxsize
        while(start <= end):
            mid = start + (end - start) // 2

            if(nums[start] == nums[mid]):
                if(key > nums[mid]):
                    key = nums[mid]
                start = start + 1
            elif(nums[end] == nums[mid]):
                if(key > nums[mid]):
                    key = nums[mid]
                end = end - 1

            elif(nums[mid] < nums[end]):
                if(key > nums[mid]):
                    key = nums[mid]
                end = mid - 1

            else:
                if(key > nums[start]):
                    key = nums[start]
                start = mid + 1

                
        
        return key

 

测试用例：

if __name__=="__main__":
    nums = [2,2,2,0,1]
    res = Solution().findMin(nums)
    print(res)