深夜小肝,整理一道华子OD一面手撕算法...
题目描述
符合下列属性的数组 arr 称为 山脉数组:
1) arr.length >= 3
2) 存在 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 ,返回任何满足以上条件的下标 i 。
示例 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
提示:
3 <= arr.length <= 104
0 <= arr[i] <= 106
题目数据保证 arr 是一个山脉数组
算法要求
- 二分法
思路分析
首先题目已规定使用二分法,则可以直接利用常规二分法的解题思路去分析该问题
如上图示,mid指针所在位置分为三种:
①指向最高山脉时:返回mid即可
②指向最高山脉左侧:即mounts[mid] > mounts[mid-1]时,在mid的右侧区间查找
③指向最高山脉右侧:即mounts[mid] < mounts[mid-1]时,在mid的左侧区间查找
代码实现
思路清晰后,代码实现就比较简单,Golang的代码实现如下:
func GetMountIndex(mounts []int, left, right int) int {
for left <= right {
mid := left + (right-left)/2
if mounts[mid] > mounts[mid-1] && mounts[mid] > mounts[mid+1] {
//mid指针在最高山脉位置
return mid
} else if mounts[mid] < mounts[mid-1] {
//mid指针在最高山脉的右侧
return getMaxMountIndex(mounts, left, mid-1)
} else if mounts[mid] > mounts[mid-1] {
//mid指针在最高山脉的左侧
return getMaxMountIndex(mounts, mid+1, right)
}
}
//未找到返回-1
return -1
}