题目描述
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。 输入一个非减排序的数组的一个旋转,输出旋转数组的最小元素。 例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。
NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
思路
《剑指offer》P66
旋转数组的本质是两个有序数组,其中后一个数组的所有元素都不大于前一个数组。因此,可以使用二分法找到两个数组的分界点。
具体步骤如下:
- 若mid位置的元素同时满足以下两个条件,则该元素则该元素属于前一个有序数组,则将左边界向mid右边移动一位。
- 大于等于左边界元素值;
- 大于等于右边界元素值。
- 若mid位置的元素小于等于右边界,则将右边界移动到mid位置处。
- 最终返回左边界位置上的数。
注意
这里有一个小小的坑,在书里也有特别说明:
[1, 0, 1, 1, 1]
[1, 1, 1, 0, 1]
对于上面的两个样例,left right mid 三个位置上的值都是1,无法判断该值属于前一个数组还是后一个,此时的结果取决于比较的判断条件哪一条写在前面,哪一条就会生效。但是实际上结果是不对的。
因此,在三个位置数字相等的情况下,无法继续使用二分法进行比较,应该使用顺序查找的方法。
但是牛客网上的测试样例中应该是没有这种特殊样例,因此即使没有考虑到这种情况,代码也是可以AC的。
code
# -*- coding:utf-8 -*-
class Solution:
def minNumberInRotateArray(self, rotateArray):
# write code here
if not rotateArray:
return 0
left = 0
right = len(rotateArray) - 1
while left < right:
mid = (left + right) / 2
print "mid : ", rotateArray[mid]
# 如果三个位置上的值相等,则只能顺序查找
if rotateArray[mid] == rotateArray[right] and rotateArray[mid] == rotateArray[left]:
return self.minInOrder(rotateArray, left, right)
if rotateArray[mid] <= rotateArray[right]:
right = mid
elif rotateArray[mid] >= rotateArray[left]:
left = mid + 1
return rotateArray[left]
def minInOrder(self, array, startIndex, endIndex):
res = array[startIndex]
for i in range(startIndex + 1, endIndex):
if array[i] < res:
res = array[i]
return res