LeetCode 154 [Find Minimum in Rotated Sorted Array II]

原題

假設一個旋轉排序的數組其起始位置是未知的(比如0 1 2 4 5 6 7 可能變成是4 5 6 7 0 1 2)。

你需要找到其中最小的元素。
數組中可能存在重復的元素。
給出[4,4,5,6,7,0,1,2] 返回 0

解題思路

  • 與[Find Minimum in Rotated Sorted Array I]相比,本題中有重復元素,還是讓nums[mid]nums[end]比較,但要考慮nums[mid] == mums[end]的情況,此時 end -= 1
  • 極端情況[1, 2, 2, 2, .....2],nums[mid]mums[end]一直相等,操作一直是end -= 1,所以時間復雜度變成O(n)
  • 所以方法二就是直接遍歷一遍,維護一個最小值
  • 對于方法一,還可以想成在普通的binary search基礎上,加入兩個while循環
while start + 1 < end:
    mid = start + (end - start) / 2
    # 如果重復,start向后移動
    while nums[mid] == nums[start] and start < mid:
        start += 1
    # 如果重復,end向前移動
    while nums[mid] == nums[end] and end > mid:
        end -= 1
    if nums[mid] > nums[end]:
        start = mid
    elif nums[mid] < nums[end]:
        end = mid

完整代碼

# 方法一
class Solution(object):
    def findMin(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return 0
            
        start, end = 0, len(nums) - 1
        while start + 1 < end:
            mid = start + (end - start) / 2
            if nums[mid] > nums[end]:
                start = mid
            elif nums[mid] < nums[end]:
                end = mid
            else:
                end -= 1
        return min(nums[start], nums[end])

# 方法二
class Solution(object):
    def findMin(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        if not nums:
            return
        min = nums[0]
        for item in nums:
            if item < min:
                min = item
        return min
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容