原題
假設(shè)一個旋轉(zhuǎn)排序的數(shù)組其起始位置是未知的(比如0 1 2 4 5 6 7 可能變成是4 5 6 7 0 1 2)。
你需要找到其中最小的元素。
數(shù)組中可能存在重復(fù)的元素。
給出[4,4,5,6,7,0,1,2] 返回 0
解題思路
- 與[Find Minimum in Rotated Sorted Array I]相比,本題中有重復(fù)元素,還是讓
nums[mid]
和nums[end]
比較,但要考慮nums[mid] == mums[end]
的情況,此時end -= 1
- 極端情況[1, 2, 2, 2, .....2],
nums[mid]
和mums[end]
一直相等,操作一直是end -= 1
,所以時間復(fù)雜度變成O(n) - 所以方法二就是直接遍歷一遍,維護一個最小值
- 對于方法一,還可以想成在普通的binary search基礎(chǔ)上,加入兩個while循環(huán)
while start + 1 < end:
mid = start + (end - start) / 2
# 如果重復(fù),start向后移動
while nums[mid] == nums[start] and start < mid:
start += 1
# 如果重復(fù),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