题目:
符合下列属性的数组 arr 称为 山峰数组(山脉数组) :
arr.length >= 3
存在 i(0 < i < arr.length - 1)使得:
arr[0] < arr[1] < … arr[i-1] < arr[i]
arr[i] > arr[i+1] > … > arr[arr.length - 1]
给定由整数组成的山峰数组 arr ,返回任何满足 arr[0] < arr[1] < … arr[i - 1] < arr[i] > arr[i + 1] > … > arr[arr.length - 1] 的下标 i ,即山峰顶部。
输入:arr = [1,3,5,4,2]
输出:2
输入:arr = [0,10,5,2]
输出:1
分析:
可以直观的看出来逐一扫描整个数组,通过比较就能找出数组中的最大值,时间复杂度为O(n),完全没利用题目中的条件。
题目中可见数组是先递增后递减的,最大值是数组中唯一一个比它左右两边数字都大的数字,位于最大值前面的数字(除第一个数字外)总是比它前一个数字大但比它后一个数字小,同理最大值后面的数字(除最后一个数字外)总是比它后一个数字大但比它前一个数字小。
可以根据这个山峰的特点来进行二分查找法,具体详见代码注释。
需要注意在一个长度为n的山峰数组中,由于第一个数字和最后一个数字都不可能是最大值,因此初始查找范围为数组下标从1到n-2的部分,如果输入的数组是一个有效的山峰数组,那么在while循环中一定能找到山峰数组的最大值,只是java语法要求函数的每个分支必须要有返回值,所以在函数体的最后一行返回-1的代码,实际上,这一行代码不会被执行。
代码:
class Solution {
public int peakIndexInMountainArray(int[] nums) {
int left = 1;
int right = nums.length-2;
while (left<=right){
int mid =(left+right)/2;
// 如果下标为mid的数字大于它前后两个数字,那么它就是最大值
if (nums[mid]>nums[mid+1]&&nums[mid]>nums[mid-1]){
return mid;
}
// 如果下标为mid的数字大于前面的数字,那么它位于数组递增的部分,接下来查找它的后半部分
if (nums[mid]>nums[mid-1]){
left = mid + 1;
// 否则它位于数组递减的部分,接下来查找它的前半部分
}else {
right = mid -1;
}
}
return -1;
}