题目

162. 寻找峰值 (中等)
峰值元素是指其值严格大于左右相邻值的元素。

给你一个整数数组 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]
  */

题解

import java.util.Scanner;

/**
 * 时间: 2021-09-15 22:13
 * 分析: 查找数组中的峰值元素
 * 思路:
 *  1. 遍历该数组, 比较当前元素与前后元素
 * 优化:
 *  1. 在 思路1 的基础上, 增加判断数组长度小于3的情况 (当前实现)
 * 参考:
 *  1. 二分法 (TODO)
 */
class Solution_0915_162 {
    public int findPeakElement(int[] nums) {
        if (nums.length == 1) {
            return 0;
        }
        if (nums[0] > nums[1]) {
            return 0;
        }
        if (nums[nums.length - 1] > nums[nums.length - 2]) {
            return nums.length - 1;
        }

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

Q.E.D.