LeetCode刷题记录——第852题（山脉数组的峰顶索引）

最新推荐文章于 2022-12-04 17:38:52 发布

原创最新推荐文章于 2022-12-04 17:38:52 发布 · 160 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

菜鸟的LeetCode刷题记录专栏收录该内容

165 篇文章

订阅专栏

题目描述

我们把符合下列属性的数组 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 的值。

示例 1：

输入：[0,1,0]
输出：1

示例 2：

输入：[0,2,1,0]
输出：1

提示：

3 <= A.length <= 10000
0 <= A[i] <= 10^6
A 是如上定义的山脉

思路分析

相当于复习一下二分查找法

代码示例

class Solution(object):
    def peakIndexInMountainArray(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        left = 0
        right = len(A)
        while left <= right:
            mid = (left + right) // 2
            mid_l = mid - 1
            mid_r = mid + 1
            if A[mid] > A[mid_l] and A[mid] > A[mid_r]:
                return mid
            elif A[mid] < A[mid_l]:
                right = mid
            elif A[mid] < A[mid_r]:
                left = mid