一、题目
把一个数组最开始的若干个元素搬到数组的末尾,我们称之为数组的旋转。输入一个递增排序的数组的一个旋转,输出旋转数组的最小元素。例如数组{3, 4, 5, 1, 2}为{1, 2, 3, 4, 5}的一个旋转,该数组的最小值为1。
二、解法
分析:这道题最直观的解法很简单那,从头到尾遍历数组一次,就得到了最小的元素,这种思路的时间复杂度显然是O(n),但这种思路没有利用输入的旋转数组的特性,达不到面试官的要求。
我们可以利用二分查找的思路解决这个问题,用两个指针分别指向数组的第一个元素和最后一个元素。按照题目中旋转的规则,第一个元素应该是大于或者等于最后一个元素的。接着我们可以找到数组中间的元素,如果中间元素大于等于第一个指针指向的元素,此时数组中最小的元素应该位于中间元素的后面,可以把第一个指针指向该中间元素,这样可以缩小寻找的范围,如果中间元素小于等于第二个指针指向的元素,此时吧第二个指针指向中间元素。不管是移动第一个指针还是第二个指针,查找范围都会缩小到原来的一半,接下来我们再用更新之后的 两个指针重复做新一轮的查找。
基于上诉思路我们有以下代码
int Min(int *numbers, int length) { if(numbers==nullptr||length<=0) throw new std:: exception("Invalid parameters!"); int index1 = 0; int index2 = length-1; int indexMid = index1; while(numbers[index1] >= numbers[index2]) { if(index2-index1==1) { indexMid = index2; break; } indexMid = (index1+index2)/2; if(numbers[indexMid] >= numbers[index1]) index1 = indexMid; else if(numbers[indexMid] <= numbers[index2]) index2 = indexMid; } return numbers[indexMid]; }
针对前面的代码,其实存在一种bug,即若原始数组为{0, 1, 1, 1, 1},其两种旋转数组可以为{1, 0, 1, 1, 1}和{1, 1, 1, 0, 1},在前面的代码中,index1和index2对应的两个数相同,并且,它们中间的数字indexMid指向的数也为1,即三个指针指向的数字都是相同的数字,在这种情况下,上面代码会出错。对于第一种旋转数组:因为numbers[indexMid] >= numbers[index1],即index1=indexMid, 错过了前面的数字0,认为最小数字在后面。
因此需要对前面的代码改进,当numbers[index1] == numbers[index2] && numbers[indexMid] == numbers[index1]时,采用顺序查找的方法:
//顺序查找 int MinInOrder(int* numbers, int index1, int index2) { int result = numbers[index1]; for(int i = index1 + 1; i <= index2; ++i) { if(result > numbers[i]) result = numbers[i]; } return result; } int Min(int* numbers, int length) { if(numbers == nullptr || length <= 0) throw new std::exception("Invalid parameters"); int index1 = 0; int index2 = length - 1; int indexMid = index1; while(numbers[index1] >= numbers[index2]) { // 如果index1和index2指向相邻的两个数, // 则index1指向第一个递增子数组的最后一个数字, // index2指向第二个子数组的第一个数字,也就是数组中的最小数字 if(index2 - index1 == 1) { indexMid = index2; break; } // 如果下标为index1、index2和indexMid指向的三个数字相等, // 则只能顺序查找 indexMid = (index1 + index2) / 2; if(numbers[index1] == numbers[index2] && numbers[indexMid] == numbers[index1]) return MinInOrder(numbers, index1, index2); // 缩小查找范围 if(numbers[indexMid] >= numbers[index1]) index1 = indexMid; else if(numbers[indexMid] <= numbers[index2]) index2 = indexMid; } return numbers[indexMid]; }