题目描述
我们把符合下列属性的数组 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
的值。
样例
输入:[0,1,0]
输出:1
输入:[0,2,1,0]
输出:1
注意
3 <= A.length <= 10000
0 <= A[i] <= 10^6
A
是如上定义的山脉。
算法
(暴力枚举) $O(n)$
- 从位置 1 开始遍历数组,发现
A[i - 1] < A[i] and A[i] > A[i + 1]
返回i
即可。
时间复杂度
- 最多遍历一次数组,故时间复杂度为 $O(n)$。
空间复杂度
- 仅使用了常数的额外空间。
C++ 代码
class Solution {
public:
int peakIndexInMountainArray(vector<int>& A) {
int n = A.size();
for (int i = 1; i < n - 1; i++)
if (A[i - 1] < A[i] && A[i] > A[i + 1])
return i;
return -1;
}
};