刷题--旋转数组的最小数字

把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。 输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。 例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。 NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。

旋转之后的数组实际上可以划分为两个排序的子数组,且前面的子数组的元素都大于或者等于后面子数组的元素。最小的元素恰好是这两个子数组的分界线。在排序的数组中我们可以用二分查找法实现O(logn)的查找。
和二分查找一样,我们用两个变量(index_start,index_end)分别指向数组的第一个元素和最后一个元素。如果中间元素位于前面的递增子数组,那么它应该大于等于index_start所指向的元素,此时数组中最小的元素应该位于中间元素的后面。我们可以把index_start指向该中间元素,缩小查找的范围。同样,如果中间元素位于后面的递增子数组,那么它应该小于等于index_end所指向的元素,此时数组中最小的元素应该位于中间元素的前面。我们可以把index_end指向该中间元素。最终index_start会指向前面子数组的最后一个元素,index_end会指向后面子数组的第一个元素,也就是说,它们最终会指向两个相邻的元素,而index_end指向的元素刚好就是数组的最小元素。

# -*- coding:utf-8 -*-
class Solution:
    def minNumberInRotateArray(self, rotateArray):
        # write code here
        if len(rotateArray) == 0:
            return 0

        index_start = 0
        index_end = len(rotateArray) - 1
        index_mid = index_start  # 考虑旋转数组与原数组一致

        while (rotateArray[index_start] >= rotateArray[index_end]):
            if (index_end - index_start == 1):
                index_mid = index_end
                break

            index_mid = (index_end + index_start) / 2

            if (rotateArray[index_mid] < rotateArray[index_end]):
                index_end = index_mid
            # 考虑类似[1, 0, 1, 1, 1],[1, 1, 1, 0, 1]的情况
            elif (rotateArray[index_mid] == rotateArray[index_end]):
                index_end -= 1
            else:
                index_start = index_mid

        return rotateArray[index_mid]

猜你喜欢

转载自blog.csdn.net/treasure_z/article/details/79940207