【剑指offer】旋转数组的最小数字 python+C++

182 阅读1分钟

【题目描述】

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

【思路】

主要用二分法,旋转后的数组将最后几位移到数组前面,一般,数组头部大于尾部,要找最小数,通过二分法缩小范围。三个指针left,mid,right。

当mid大于left,说明最小值在mid之后,更新left=mid;

当mid小于right,说明最小值在mid之前,更新right=mid;

但是有两种特殊情况,当旋转值为0,直接返回left;当mid==left==right,只能遍历找最小值。

【代码】

C++:
class Solution {
public:
    int minNumberInRotateArray(vector<int> rotateArray) {
        int left=0;
        int right=rotateArray.size()-1;
        int mid=int((left+right)/2);
        if(rotateArray[left]<rotateArray[right]){
            return rotateArray[left];
        }
        
        if(rotateArray[mid]==rotateArray[left]&&rotateArray[left]==rotateArray[right]){
            int temp=rotateArray[0];
            for(int i=1;i<=right;i++){
                if(rotateArray[i]<temp){
                    temp=rotateArray[i];
                }
            }
            return temp;
        }
        
        while(right-left>1){
            mid=int((left+right)/2);
            if(rotateArray[mid]<=rotateArray[right]){
                right=mid;
            }else if(rotateArray[mid]>=rotateArray[left]){
                left=mid;
            }
        }
        return rotateArray[right];
    }
};
python:
# -*- coding:utf-8 -*-
class Solution:
    def minNumberInRotateArray(self, rotateArray):
        # write code here
        left=0
        right=len(rotateArray)-1
        mid=(left+right)//2
        if(rotateArray[left]<rotateArray[right]):
            return rotateArray[left]
        if (rotateArray[left]==rotateArray[right] and rotateArray[mid]==rotateArray[left]):
            return min(rotateArray)
        while(right-left>1):
            mid=(left+right)//2
            if rotateArray[mid]>=rotateArray[left]:
                left=mid
            elif(rotateArray[mid]<=rotateArray[right]):
                right=mid
        return rotateArray[right]