852. 山脉数组的峰顶索引
我们把符合下列属性的数组 A 称作山脉:
A.length >= 3
存在 0 < i < A.length - 1 使得A[0] < A[1] < … A[i-1] < A[i] > A[i+1] > … > A[A.length - 1]
给定一个确定为山脉的数组,返回任何满足 A[0] < A[1] < … A[i-1] < A[i] > A[i+1] > … > A[A.length - 1] 的 i 的值。
示例 1:
输入:[0,1,0]
输出:1
示例 2:
输入:[0,2,1,0]
输出:1
提示:
3 <= A.length <= 10000
0 <= A[i] <= 10^6
A 是如上定义的山脉
解题思路:
1.用库函数,max 和index
2.用二分查找方法进行快速查找
代码:
class Solution(object):
def peakIndexInMountainArray(self, A):
"""
:type A: List[int]
:rtype: int
"""
#直接借助函数
# return A.index(max(A))
#二分法实现查找最大值
left = 0
right = len(A)-1
while left <= right:
mid = (right+left)//2
if A[mid]>A[mid+1] and A[mid]>A[mid-1]:
return mid
elif A[mid]<A[mid+1] and A[mid]>A[mid-1]:
left = mid + 1
else:
right = mid - 1
return -1