leetcode 算法-寻找峰值-162

leetcode 算法-寻找峰值

leetcode 传送门

题目

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

给定一个输入数组 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。

解题思路

本题是寻找一个数组中的峰值元素,即峰值元素大于左右元素,需要注意的是，如果在边界,只需要大于有元素的一边即可。
这道题也比较简单,一个很明显也很暴力的的想法,先判断首部元素,则判断是否大于第二个元素,如果大于则直接返回0;
然后判断尾部元素,如果大于上一个元素,则直接返回数组长度-1;然后从1开始遍历数组,如果满足当前元素值大于左右元素值,则直接返回即可。这种方法简单粗暴,但是速度并不快,时间复杂度$O(n)$级。

接下来,我们进入正题,介绍第二种方法,用熟悉的二分法解这道题。

设数组为nums,尾部索引为end,中间索引为mid。

本题是目标是用最少的步骤的寻找到波峰,波峰的特点是左边是上坡,右边是下坡。
假如nums[mid]<nums[mid+1],那么nums[mid]一定是在一个上坡,那么只要有上坡,就一定会有波峰在mid的右边。
又因为nums[i] ≠ nums[i+1],所以nums[mid]≠nums[mid+1]
同理,如果nums[mid]>nums[mid+1],那么波峰一定在mid的左边,但是这里要注意,这里的左边是包括mid本身的。因为mid本身是波峰,也可以满足nums[mid]>nums[mid+1]。

经过上面的步骤,我们就可以缩小一半的范围,然后不断迭代缩小波峰的范围,最终就可以得出波峰的位置所在。

代码

class Solution {
    public int findPeakElement(int[] nums) {
        if (nums == null || nums.length < 2) return 0;
        int right = nums.length-1, mid = right >> 1, left = 0;
        while (left < right) {
            mid = ((left+right) >> 1);
            if (nums[mid] > nums[mid+1]) {//左半部分有波峰
                right = mid;
            } else{//右半部分有波峰
                left = mid + 1;
            }
        }
        return left;
    }
}