寻找峰值

162. 寻找峰值

难度中等596

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

给你一个整数数组 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。

提示：

• 1 <= nums.length <= 1000
• -231 <= nums[i] <= 231 - 1
• 对于所有有效的 i 都有 nums[i] != nums[i + 1]

精炼的写法

class Solution {
    public int findPeakElement(int[] nums) {
        int l = 0, r = nums.length - 1;
        while (l < r) {
            int mid = l + (r - l) / 2;
            if (nums[mid] > nums[mid + 1])
                // 这里是核心，如果找到一个值大于右边的值，那么不往左一个
                r = mid;
            else
                // 如果一个值大于右边的值
                l = mid + 1;
        }
        return l;
    }
}

笨方法

class Solution {
    public int findPeakElement(int[] nums) {
        int n = nums.length;
        int l = 0,r=n-1;
        if(n==1)return 0;
        if(n==2)return nums[0]<nums[1]?1:0;

        while(l<r){
            int m = l+(r-l)/2;
            int left = m-1;
            int right = m+1;
            if(left>=0&&right<n&&nums[m-1]<nums[m]&&nums[m]>nums[m+1]){
                return m;
            }else if(left>=0&&nums[m-1]<nums[m]){
                l = m+1;
            }else if(right<n&&nums[m]>nums[m+1]){
                r = m-1;
            }else{
                l++;
            }
        }
        return l;
    }
}

赏

谢谢你请我吃糖果

微信