力扣题库162. 寻找峰值

问题描述

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

给你一个整数数组 nums，找到峰值元素并返回其索引。数组可能包含多个峰值，
在这种情况下，返回 任何一个峰值 所在位置即可。

你可以假设 nums[-1] = nums[n] = -∞ 。

你必须实现时间复杂度为 O(log 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++代码片

下面展示为力扣中的C++风格的 代码片.

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int left=0,right=nums.size()-1;
        int res=INT_MIN;
        for(int i=0;i<=right;i++)
        {
            res=max(nums[i],res);
            //优化再次使用for循环来获取下标，使时间复杂度：从2n变成n
            if(nums[i]==res)//更新left下标
            {
                left=i;
            }
        }
        /*for(int i=0;i<=right;i++)
        {
        	if(nums[i]==res)//更新left下标
            {
                left=i;
            }
        }*/
        return left;
    }
};

复杂度分析

时间复杂度：O(n)，其中 n是数组 nums 的长度。
空间复杂度：O(1)。

由于题目中必须实现时间复杂度为 O(log n) 的算法来解决。
故而我们可以用二分法查找来找最大值。

下面展示为力扣中的C++风格的 代码片.

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int left=0,right=nums.size()-1;
        while(left<right)
        {
            int mid = left + (right - left) / 2;
            if(nums[mid]>nums[mid+1]) 
            {
                // 峰值在左侧，包括mid
                right=mid;
            }
            else
            {
                // 峰值在右侧，不包括mid
                left=mid+1;
            }
        }
        return left;
    }
};

复杂度分析

时间复杂度：O(log n) ，其中 n是数组 nums 的长度。
空间复杂度：O(1)。