剑指 Offer II 069. 山峰数组的顶部

剑指 Offer II 069. 山峰数组的顶部

符合下列属性的数组 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/

思路

常规二分法解即可，两侧设为-Infinity，山顶总是在上坡的那一侧

解答

/**
 * @param {number[]} arr
 * @return {number}
 */
var peakIndexInMountainArray = function (arr) {
  let left = 0,
    right = arr.length - 1;
  while (left <= right) {
    const mid = parseInt((left + right) / 2);
    const leftVal = mid > 0 ? arr[mid - 1] : -Infinity;
    const rightVal = mid < arr.length - 1 ? arr[mid + 1] : -Infinity;
    if (arr[mid] > leftVal && arr[mid] > rightVal) {
      return mid;
    } else if (rightVal > leftVal) {
      left = mid + 1;
    } else {
      right = mid - 1;
    }
  }
};