我们把符合下列属性的数组 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 是如上定义的山脉
解题思路:循环找到左值小于中间值,中间值大于右值的数。折半查找最快
方法一、循环
class Solution {
public:
int peakIndexInMountainArray(vector<int>& A) {
for(int i=0;i<A.size()-1;i++){
if(A[i]>A[i+1])return i;
}
return A.size()-1;
}
};
方法二、折半查找
class Solution {
public:
int peakIndexInMountainArray(vector<int>& A) {
int l=0, r=A.size()-1;
while(l<r) {
int m = l + (r-l)/2;
if(A[m]>A[m-1] && A[m]>A[m+1]) return m;
else if(A[m]>A[m+1]) r=m-1;
else l=m+1;
}
return l;
}
};