我们把符合下列属性的数组 A
称作山脉:
A.length >= 3
- 存在
0 < i < A.length - 1
使得A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1]
给定一个确定为山脉的数组,返回任何满足 A[0] < A[1] < ... A[i-1] < A[i] > A[i+1] > ... > A[A.length - 1]
的 i
的值。
示例 1:
输入:[0,1,0] 输出:1
示例 2:
输入:[0,2,1,0] 输出:1
提示:
3 <= A.length <= 10000
- 0 <= A[i] <= 10^6
- A 是如上定义的山脉
1、由题目可知,给定的数组A已经确定为一个山脉数组,所以不用去判断数组是否为山脉数组;
2、只需要找到一个元素符合 A[ i - 1] < A[i] > A[i + 1]即可
第一种最简单最容易想到的方法:直接遍历查找
public int peakIndexInMountainArray(int[] A) {
int index = 0;
for(int i = 1 ; i < A.length -1 ; i++) {
if(A[i] > A[i - 1] && A[i] > A[i + 1]) {
index = i ;
break ;
}
}
return index ;
}
第二种:二分查找
public int peakIndexInMountainArray(int[] A) {
int low = 0 , high = A.length - 1;
int mid = ( low + high ) / 2 ;
while( mid <= high && mid > 0 && mid >= low ) {
if( A[mid] > A[mid - 1] && A[mid] > A[mid +1])
return mid;
else if( A[mid] < A[mid - 1])
high = mid - 1 ;
else if( A[mid] < A[mid + 1])
low = mid + 1 ;
mid = ( low + high ) / 2 ;
}
return A[mid] > A[mid + 1] ? mid : mid + 1 ;
}