[Golang]力扣Leetcode - 852. 山脉数组的峰顶索引（二分查找）

题目：符合下列属性的数组 arr 称为 山脉数组 ：

• arr.length >= 3

• 存在 i（0 < i < arr.length - 1）使得：

  • arr[0] < arr[1] < ... arr[i-1] < arr[i]
  • arr[i] > arr[i+1] > ... > arr[arr.length - 1]

给你由整数组成的山脉数组 arr ，返回任何满足 arr[0] < arr[1] < ... arr[i - 1] < arr[i] > arr[i + 1] > ... > arr[arr.length - 1] 的下标 i 。

链接： 力扣Leetcode - 852. 山脉数组的峰顶索引.

示例 1：

输入：arr = [0,1,0]
输出：1

示例 2：

输入：arr = [0,2,1,0]
输出：1

示例 3：

输入：arr = [0,10,5,2]
输出：1

示例 4：

输入：arr = [3,4,5,1]
输出：2

示例 5：

输入：arr = [24,69,100,99,79,78,67,36,26,19]
输出：2

思路：定义查找的范围 [left,right]，初始查找范围是整个数组。每次取查找范围的中点 mid，比较 nums[mid] 和 左右边的 nums[mid-1] 和 nums[mid+1] 的大小，如果刚好比两边都大则 mid 即为要寻找的下标，如果不相等则根据 nums[mid] 和 nums[mid-1] 、 nums[mid+1] 的大小关系将查找范围缩小一半。

• 如果 arr[mid] > arr[mid-1] && arr[mid] > arr[mid+1]，则下标 mid 即为要寻找的下标；
• 如果 arr[mid] < arr[mid-1]，则山峰只可能在下标 mid 的左侧，right = mid - 1；
• 如果 arr[mid] > arr[mid-1]，则山峰只可能在下标 mid 的右侧，left = mid + 1

Go代码：

package main

import "fmt"

func peakIndexInMountainArray(arr []int) int {
    left, right := 1, len(arr)-2
    for left <= right {
        mid := (left + right) / 2
        if arr[mid] > arr[mid-1] && arr[mid] > arr[mid+1] {
            return mid
        } else if arr[mid] < arr[mid-1] {
            right = mid - 1
        } else {
            left = mid + 1
        }
    }
    return right
}

func main() {
    a := []int{0, 2, 1, 0}
    fmt.Println(peakIndexInMountainArray(a))
}

提交截图：