算法题解:旋转数组的最小数字
题目描述
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。
解题思路
将旋转数组对半分可以得到一个包含最小元素的新旋转数组,以及一个非递减排序的数组。新的旋转数组的数组元素是原数组的一半,从而将问题规模减少了一半,这种折半性质的算法的时间复杂度为 O(logN)(为了方便,这里将 log2N 写为 logN)。
此时问题的关键在于确定对半分得到的两个数组哪一个是旋转数组,哪一个是非递减数组。我们很容易知道非递减数组的第一个元素一定小于等于最后一个元素。
通过修改二分查找算法进行求解:
- 当 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 在哪个区间。
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
版权声明:凡未经本网站书面授权,任何媒体、网站及个人不得转载、复制、重制、改动、展示或使用本网站的局部或全部的内容或服务,或在非本网站所属服务器上建立镜像。如果已转载,请自行删除。同时,我们保留进一步追究相关行为主体的法律责任的权利。我们希望与各媒体合作,签订著作权有偿使用许可合同,故转载方须书面/邮件申请,以待商榷。