LeetCode 162. 寻找峰值

【162. 寻找峰值】

题目描述

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

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

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

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

注意提示里的信息。

解法1：寻找最大值

我们最先想到的暴力方法一定是去遍历数组，找到数组中的最大值，就是这个数组的峰值。

显然这个解法是不符合题目中时间复杂度为 O(log n) 的条件的。
代码如下：

class Solution {
    public int findPeakElement(int[] nums) {
        int idx = 0;
        for (int i = 1; i < nums.length; ++i) {
            if (nums[i] > nums[idx]) {
                idx = i;
            }
        }
        return idx;
    }
}

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

解法2：二分查找爬坡法

看到题目要求时间复杂度为O(log(n))，我们很容易想到能不能用二分法。

对于这道题，我们可以利用二分法，对二分得到的mid下标的元素 nums[mid] 和 nums[mid+1] 比较大小，如果 nums[mid] < nums[mid+1]，说明我们一定能找到一个峰值在mid右边，令 l = mid + 1 ，继续二分；反之，则可以认为当前mid下标元素为峰值或者峰值元素在左边，令 r = mid ，继续二分查找。最后得到的r即为峰值下标。

class Solution {
    public int findPeakElement(int[] nums) {
        // 二分法
        int l = 0;
        int r = nums.length - 1;
        while (l < r) {
            int mid = (l + r) / 2;
            if (nums[mid] < nums[mid+1]) {
                l = mid + 1;
            } else {
                r = mid;
            }
        }
        return r;
    }
}