欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

查找旋转数组的最小元素

程序员文章站 2022-07-12 10:37:01
...
  • 题目要求:
    把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。
    输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。
    例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。
    NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
    
  • 解题思路:
  1. 数组为空,返回0
  2. 数组未旋转(原排列),或数组长度为1,返回第一个元素
  3. 正常情况下可以采用以下两种方案
# 1. 容易想到:利用非递减的特点,找到数组中大于相邻下一个元素的值即为最小值
# 不适用数据较多且类似[2,3,4,5,1]的情况
class Solution1:
    def minNumberInRotateArray(self, rotateArray):
        # write code here
        if not rotateArray:
            return 0
        for i in range(len(rotateArray) - 1):
            if rotateArray[i] > rotateArray[i + 1]:
                return rotateArray[i + 1]
        return rotateArray[0]


# 采用二分法
class Solution2:
    def minNumberInRotateArray(self, rotateArray):
        if len(rotateArray) == 0:
            return 0
        if len(rotateArray) == 1:
            return rotateArray[0]
        left = 0
        right = len(rotateArray) - 1
        while left < right:
            # 未旋转
            if rotateArray[left] < rotateArray[right]:
                return rotateArray[left]
            mid = (right + left) // 2
            if rotateArray[mid] < rotateArray[right]:
                right = mid
            elif rotateArray[left] < rotateArray[mid]:
                left = mid
            else:
                left += 1
        return rotateArray[right]


if __name__ == '__main__':
    s = Solution2()
    print(s.minNumberInRotateArray([3, 4, 5, 1, 2]))
相关标签: 数据结构与算法