162. 寻找峰值【模拟】

162. 寻找峰值

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
• -2^31 <= nums[i] <= 2^31 - 1
• 对于所有有效的 i 都有 nums[i] != nums[i + 1]

解题思路

直接模拟法：考虑好一些边界情况即可

func findPeakElement(nums []int) int {
    // 题目表明切片元素至少一个，如果只有一个，则这个唯一的数就是峰值，因为两边是负无穷
    if len(nums) == 1 {
        return 0
    }
    // 考虑边界情况，第一个和最后一个元素
    // 通过上面的if后，可以知道元素至少有两个了
    if nums[0] > nums[1] {
        return 0
    }
    if nums[len(nums)-1] > nums[len(nums)-2] {
        return len(nums) -1
    }
    for i := 1;i < len(nums) - 1;i++ {
        if nums[i] > nums[i-1] && nums[i] > nums[i+1] {
            return i
        }
    }
    return -1
}