leetcode--852--山脉数组的峰顶索引

2020-07-19  本文已影响0人  minningl

题目:
我们把符合下列属性的数组 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 是如上定义的山脉

链接:https://leetcode-cn.com/problems/peak-index-in-a-mountain-array

思路:
1、这道题和162题类似
2、使用左右2个节点,还有一个mid中间节点。如果mid位置的值比mid+1大,则说明mid节点可能是右侧的极大值,所以right=mid。否则如果mid位置的值不比mid+1的大,则说明mid节点不是右侧的极大值,需要继续往右走,即left=mid+1,为什么这里需要+1呢?因为此时mid位置不是极大值点

Python代码:

class Solution(object):
    def peakIndexInMountainArray(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        left = 0
        right = len(A)-1

        while left<right:
            mid = (left+right)/2
            if(A[mid]>A[mid+1]):
                right = mid
            else:
                left = mid+1
        return left

C++代码:

class Solution {
public:
    int peakIndexInMountainArray(vector<int>& A) {
        int left=0;
        int right=A.size()-1;

        while(left<right){
            int mid=(left+right)/2;
            if(A[mid]>A[mid+1]){
                right = mid;
            }else{
                left = mid+1;
            }
        }
        return left;
    }
};
上一篇下一篇

猜你喜欢

热点阅读