162. 寻找峰值

小豆丁 1年前 ⋅ 798 阅读
162. 寻找峰值
峰值元素是指其值大于左右相邻值的元素。

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

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

 

示例 1:

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

解析

第一种方法我们可以直接进行线性遍历,只需要找到前一个是否大于后一个即可。

可以三种情况都进行讨论,一种是一直向上,第二种是一直向下,第三种是由转折。我们发现条件都是满足的。

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

logN的方法

一般logN都是二分查找。

这里我们可以利用mid 与 mid+1 的关系去确定处于上生态还是下降态,最终只剩下一个元素,这就是我们要找的顶部。

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