山峰数组的顶部
题目描述:
符合下列属性的数组 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 ,即山峰顶部。
示例 :
输入:arr = [0,1,0]
输出:1
提示:
- 3 <= arr.length <= 104
- 0 <= arr[i] <= 106
- 题目数据保证 arr 是一个山脉数组
解法
二分法
代码
class Solution:
def peakIndexInMountainArray(self, arr: List[int]) -> int:
left, right = 0, len(arr) - 1
while left <= right:
mid = left + (right - left) // 2
if arr[mid - 1] > arr[mid] > arr[mid + 1]:
right = mid - 1
elif arr[mid - 1] < arr[mid] < arr[mid + 1]:
left = mid + 1
else:
return mid
测试结果
执行用时:24 ms, 在所有 Python3 提交中击败了 97.19% 的用户
内存消耗:15.8 MB, 在所有 Python3 提交中击败了 65.86% 的用户
说明
算法题来源:力扣(LeetCode)