首页 > 技术文章 > 寻找峰值 -- LeetCode -- 9.15

rongrongrong 2021-09-15 11:03 原文

寻找峰值

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

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

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

  你必须实现时间复杂度为 O(log n) 的算法来解决此问题。

示例 1:

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

经典二分查找问题:

  因为它的最左最右边是 负无穷,所以说一定存在峰值。

if nums[mid] < nums[mid + 1]
    //说明 mid 点后方一定存在峰值,否则为前方

  接下来就是考虑 二分查找模板问题了;

  如果是  if nums[mid] < nums[mid + 1] 这种情况,说明当前 mid 点肯定不是咱要找的点,l = mid + 1就很好理解;否则的话,就是说,峰值在 mid 前面,而 mid 点可能就是个峰值,所以说 r = mid 而不能 r = mid - 1;

代码:

class Solution {
public:
    int findPeakElement(vector<int>& nums) {
        int l = 0, r = nums.size() - 1;
        while (l < r) {
            int mid = l + ((r - l) >> 1);
            if (nums[mid] < nums[mid+1]) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        return r;//最后推出循环条件是 l == r 这里返回那个都无所谓
    }
};

  

  

推荐阅读