算法题解:旋转数组的最小数字

题目描述

把一个数组最开始的若干个元素搬到数组的末尾,咱们称之为数组的旋转。输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。java

解题思路

将旋转数组对半分能够获得一个包含最小元素的新旋转数组,以及一个非递减排序的数组。新的旋转数组的数组元素是原数组的一半,从而将问题规模减小了一半,这种折半性质的算法的时间复杂度为 O(logN)(为了方便,这里将 log2N 写为 logN)。算法

此时问题的关键在于肯定对半分获得的两个数组哪个是旋转数组,哪个是非递减数组。咱们很容易知道非递减数组的第一个元素必定小于等于最后一个元素。数组

经过修改二分查找算法进行求解:code

  • 当 nums[mid] <= nums[high] 时,表示 [mid, high] 区间内的数组是非递减数组,[low, mid] 区间内的数组是旋转数组,此时令 high = mid;
  • 不然 [mid + 1, high] 区间内的数组是旋转数组,令 low = mid + 1。
public int minNumberInRotateArray(int[] arr) {
    if (arr.length == 0) {
        return 0;
    }
    int low = 0;
    int high = arr.length - 1;
    while (low < high) {
        int mid = low + (high - low) / 2;
        // 注意此处比较的是后半段
        if (arr[mid] <= arr[high]) { 
            high = mid;
        } else {
            low = mid + 1;
        }
    }
    return arr[low];
}

若是数组元素容许重复,会出现一个特殊的状况:nums[low] == nums[mid] == nums[high],此时没法肯定解在哪一个区间,须要切换到顺序查找。例如对于数组 {1, 1, 1, 0, 1},low、mid 和 high 指向的数都为 1,此时没法知道最小数字 0 在哪一个区间。blog

public int minNumberInRotateArray(int[] arr) {
    if (arr.length == 0) {
        return 0;
    }
    int low = 0;
    int high = arr.length - 1;
    while (low < high) {
        int mid = low + (high - low) / 2;
        if (arr[low] == arr[mid] && arr[mid] == arr[high]) {
            return minNumber(arr, low, high);
        } else if (arr[mid] <= arr[high]) {
            high = mid;
        } else {
            low = mid + 1;
        }
    }
     return arr[low];
}

private int minNumber(int[] arr, int low, int high) {
    for (int i = low; i < high; i++) {
        if (arr[i] > arr[i + 1]) {
            return arr[i + 1];
        }
    }
    return arr[low];
}

参考

CyC2018排序

相关文章
相关标签/搜索