LeetCode 162. 寻找峰值(C++、python)

  • Post author:
  • Post category:python


峰值元素是指其值大于左右相邻值的元素。

给定一个输入数组

nums

,其中

nums[i] ≠ nums[i+1]

,找到峰值元素并返回其索引。

数组可能包含多个峰值,在这种情况下,返回任何一个峰值所在位置即可。

你可以假设

nums[-1] = nums[n] = -∞


示例 1:

输入: nums = [1,2,3,1]
输出: 2
解释: 3 是峰值元素,你的函数应该返回其索引 2。


示例 2:

输入: nums = [1,2,1,3,5,6,4]
输出: 1 或 5 
解释: 你的函数可以返回索引 1,其峰值元素为 2;
     或者返回索引 5, 其峰值元素为 6。

C++

class Solution {
public:
    int findPeakElement(vector<int>& nums) 
    {
        int n=nums.size();
        int left=0;
        int right=n-1;
        while(left<right)
        {
            int mid=(left+right)/2;
            if(nums[mid]<nums[mid+1])
            {
                left=mid+1;
            }
            else
            {
                right=mid;
            }
        }
        return left;
    }
};

python

class Solution:
    def findPeakElement(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        n=len(nums)
        left=0
        right=n-1
        while left<right:
            mid=(left+right)//2
            if nums[mid]<nums[mid+1]:
                left=mid+1
            else:
                right=mid
        return left
        



版权声明:本文为qq_27060423原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。