题目 :合乎下列属性的数组 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))
}
提交截图: