山峰数组的顶部

山峰数组的顶部

符合下列属性的数组 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 ,即山峰顶部。

示例 1:

输入:arr = [0,1,0]
输出:1
示例 2:

输入:arr = [1,3,5,4,2]
输出:2
示例 3:

输入:arr = [0,10,5,2]
输出:1
示例 4:

输入:arr = [3,4,5,1]
输出:2
示例 5:

输入:arr = [24,69,100,99,79,78,67,36,26,19]
输出:2

提示:

3 <= arr.length <= 104
0 <= arr[i] <= 106
题目数据保证 arr 是一个山脉数组

进阶:很容易想到时间复杂度 O(n) 的解决方案,你可以设计一个 O(log(n)) 的解决方案吗?

注意:本题与主站 852  题相同:https://leetcode-cn.com/problems/peak-index-in-a-mountain-array/

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/B1IidL
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

题目已经保证 arr 是一个山脉数组了,结果肯定是在 [1, arr.length - 2] 之间。

可以用二分法,找一个中间的数 arr[mid],判断它和 arr[mid - 1], arr[mid + 1] 的关系来决定应该选择左边,还是选择右边。

代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
/**
* @param {number[]} arr
* @return {number}
*/
var peakIndexInMountainArray = function (arr) {
let lo = 1;
let hi = arr.length - 2;
while (lo < hi) {
const mid = lo + ((hi - lo) >> 1);
if (arr[mid - 1] < arr[mid] && arr[mid] > arr[mid + 1]) {
return mid;
} else if (arr[mid - 1] < arr[mid]) {
lo = mid + 1;
} else {
hi = mid - 1;
}
}
return lo;
};

作者:scnu_evan
链接:https://leetcode-cn.com/problems/B1IidL/solution/ologn-javascript-jian-zhi-offer-ii-069-s-66rl/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

__END__