把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个递增排序的数组的一个旋转,输出旋转数组的最小元素。例如,数组 [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
算法思想:
思路与算法
一个包含重复元素的升序数组在经过旋转之后,可以得到下面可视化的折线图:
其中横轴表示数组元素的下标,纵轴表示数组元素的值。图中标出了最小值的位置,是我们需要旋转的目标。
我们考虑数组中的最后一个元素 x:在最小值右侧的元素,它们的值一定都小于等于 x;而在最小值左侧的元素,它们的值一定都大于等于 x。因此,我们可以根据这一条性质,通过二分查找的方法找出最小值。
在二分查找的每一步中,左边界为low,右边界为high,区间的中点为pivot,最小值就在该区间内。我们将中轴元素 numbers[pivot] 与右边界元素numbers[high] 进行比较,可能会有以下的三种情况:
第一种情况是numbers[pivot]<numbers[high]。如下图所示,这说明 numbers[pivot] 是最小值右侧的元素,因此我们可以忽略二分查找区间的右半部分。
第二种情况是numbers[pivot]>numbers[high]。如下图所示,这说明 numbers[pivot] 是最小值左侧的元素,因此我们可以忽略二分查找区间的左半部分。
第三种情况是numbers[pivot]==numbers[high]。如下图所示,由于重复元素的存在,我们并不能确定 numbers[pivot] 究竟在最小值的左侧还是右侧,因此我们不能莽撞地忽略某一部分的元素。我们唯一可以知道的是,由于它们的值相同,所以无论 numbers[high] 是不是最小值,都有一个它的「替代品」numbers[pivot],因此我们可以忽略二分查找区间的右端点。
当二分查找结束时,我们就得到了最小值所在的位置。
def minArray(numbers) -> int:
low = 0
high = len(numbers)-1
while low < high: # 当low==high时找到最小值,注意于二分查找的对比,二分查找此处是low<=high
mid = (low + high) >> 1 # 计算中间位置
if numbers[mid] < numbers[high]:
# mid位于最小值右边
high = mid
elif numbers[mid] > numbers[high]:
# mid位于最小值左边
low = mid + 1
else:
# mid可能位于最小值右边,也可能位于最小值左边
high -= 1
return numbers[low] # 返回最小值
if __name__ == '__main__':
print(minArray([3, 4, 5, 1, 2]))
"""
运行结果:
1
Process finished with exit code 0
"""