这个题是《剑指offer》上的第八题。
牛客网题目链接:点击直达
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。
输入一个非递减排序的数组的一个旋转,输出旋转数组的最小元素。
例如数组{3,4,5,1,2}为{1,2,3,4,5}的一个旋转,该数组的最小值为1。
NOTE:给出的所有元素都大于0,若数组大小为0,请返回0。
一般思路
拿到这道题最直观的做法就是遍历一遍数组,但这种做法的时间复杂度为O(n),而且没有利用到旋转数组的特性。
由于旋转之前数组是递增的,所以旋转之后,数组实际上也是已经划分成两个已经排序好的子数组。而且前面的递增子数组的元素都大于或者等于后面子数组的元素。最小的元素恰好是这两个子数组的分界线。
我们都知道在排好序的数组中可以使用二分查找来实现O(logn)的查找,这道题数组一定程度上也是排好序的,所以可以尝试用二分查找来解决问题。
优化思路
1.定义两个指针,分别指向数组的首尾元素。按照题目中旋转规则,第一个元素应该是大于或者等于最后一个元素的。
2.第二步我们就可以找到数组中间的元素。此时分两种情况:
- 如果中间元素大于第一个指针指向的元素,说明它还在前一个递增数组中。那么最小元素一定位于它后面,此时我们将第一个指针指向该中间元素;
- 如果中间元素小于最后一个指针指向的元素,说明它在后一个递增数组中。那么最小元素应该在它前面,此时我们将第二个指针指向该中间元素。
3.按照上面的思路,第一个指针总是指向前面递增数组中的元素,第二个指针总是指向后面一个递增数组中的元素。最终第一个指针会指向前一个递增数组的最后一个元素,后一个指针会指向后面递增数组的第一个元素,也就是我们要找的最小值。
书上给的思路图:
代码
import java.util.ArrayList;
public class Solution {
public int minNumberInRotateArray(int [] array) {
int left = 0;
int right = array.length-1;
int mid=left; //初始化为left是为了防止旋转0个元素,也就是旋转后是数组本身这种情况。此时数组中第一个数字就是最小的数字,可直接返回。一旦发现数组中第一个数字小于最后一个数字,就表明数组是排好序的,就可以直接返回第一个数字。
while (array[left] >= array[right]) {
if(right - left == 1) {
mid = right;
break;
}
mid = (left+right)/2;
if(array[mid] >= array[left]) {
left = mid;
}else if(array[mid] <= array[right]) {
right = mid;
}
}
return array[mid];
}
}
拓展
上面的代码已经可以在OJ上通过了,但是这个代码遇到某些特殊的情况便不适用了,例如书上就给出了两种特殊的情况。
在上面两种情况中,第一个指针和第二个指针指向的数字都是1,并且两个指针中间的数字也是1。
在第一种情况中两个指针中间数字位于后面的子数组,而第二种情况中中间数字位于前面的子数组中。
所以遇到这种情况我们无法判断中间的数字位于前面的子数组中还是后面的子数组中。也就无法通过移动指针来缩小查找范围,此时不得不采取顺序查找的方式。
所以我们需要在代码中添加对这种这种特殊情况的判断。
代码:
import java.util.ArrayList;
public class Solution {
public int minNumberInRotateArray(int [] array) {
int left = 0;
int right = array.length-1;
int mid=left;
while (array[left] >= array[right]) {
if(right - left == 1) {
mid = right;
break;
}
mid = (left+right)/2;
//遇到我们上面说的特殊情况就采取顺序查找。
if(array[left] == array[right] && array[mid] == array[left]) {
return order(array,left,right);
}
if(array[mid] >= array[left]) {
left = mid;
}else if(array[mid] <= array[right]) {
right = mid;
}
}
return array[mid];
}
//顺序查找
public static int order(int[] array,int left,int right) {
int result = array[left];
for (int i = left + 1; i <= right; i++) {
if(result > array[i]) {
result = array[i];
}
}
return result;
}
}