LeetCode162题 - 寻找峰值

题目

峰值元素是指其值严格大于左右相邻值的元素。
给你一个整数数组 nums，找到峰值元素并返回其索引。数组可能包含多个峰值，在这种情况下，返回 任何一个峰值 所在位置即可。
你可以假设 nums[-1] = nums[n] = -∞ 。

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-peak-element
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

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

解题

题目要求时间复杂度为 O(log n)，且本题是在数组内查找要求的值，所以自然联想到二分法。

• 如果mid所在值大于左边（由于题目假设 nums[-1] = -∞，所以当mid是0是也可以），且mid所在值大于右边（由于题目假设 nums[n] = -∞，所以当mid是nums.length-1是也可以），则可以直接找到。
• 如果左边大就往左找（因为nums[-1] = -∞，所以一定能找到一个峰值）。
• 如果右边大就往右找（因为nums[n] = -∞，所以一定能找到一个峰值）。

代码

时间复杂度为 O(log n)

var findPeakElement = function(nums) {
    let left = 0,right = nums.length-1;
    while(left<right){
        const mid = Math.floor((left+right)/2);
        // 如果mid所在值大于左边（由于题目假设 nums[-1] = -∞，所以当mid是0是也可以）
        // 且mid所在值大于右边（由于题目假设 nums[n] = -∞，所以当mid是nums.length-1是也可以）
        // 则可以直接找到
        if((nums[mid] > nums[mid-1] || mid == 0) && (nums[mid]>nums[mid+1]||mid == nums.length-1)){
            return mid;
        }else if(nums[mid] < nums[mid-1]){
            // 左边大就往左找，优先往左；因为nums[-1] = -∞，所以一定能找到一个峰值
            right = mid;
        }else{
            // 右边大，就往右走
            left = mid+1;
        }
    }
    // 此时 left == right，返回谁都一样
    return right;
};