leetcode_852 山脉数组的峰顶索引

81 阅读1分钟

要求

符合下列属性的数组 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 = [0,2,1,0]
输出:1

示例 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 是一个山脉数组

核心代码

class Solution:
    def peakIndexInMountainArray(self, arr: List[int]) -> int:
        return arr.index(max(arr))

另一解法

class Solution:
    def peakIndexInMountainArray(self, arr: List[int]) -> int:
        left= 0
        right = len(arr) -1
        while left <= right:
            mid  = left + (right - left) // 2
            if arr[mid - 1] < arr[mid] < arr[mid + 1]:
                left = mid + 1
            elif arr[mid - 1] > arr[mid] > arr[mid + 1]:
                right = mid - 1
            else:
                return mid

image.png

解题思路:第一种解法:题目中有一个题目数据保证 arr 是一个山脉数组,所以我们直接获取最大值的索引即可;第二种解法:我们使用二分查找的方式,找到那个高于左边和右边的数字的索引。