题目描述:
符合下列属性的数组 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 。
我们根据题目描述可以知道峰顶是唯一的。
我自己的解法:
class Solution:
def peakIndexInMountainArray(self, arr: list) -> int:
left = 0
right = len(arr) - 1
while left <= right:
mid = (right - left) // 2 + left
if arr[mid] >= arr[mid+1] and arr[mid] >= arr[mid-1]:
return mid
elif arr[mid] < arr[mid+1]:
left = mid + 1
elif arr[mid] < arr[mid-1]:
right = mid - 1
官方的解法:
class Solution:
def peakIndexInMountainArray(self, arr: List[int]) -> int:
n = len(arr)
left, right, ans = 1, n - 2, 0
while left <= right:
mid = (left + right) // 2
if arr[mid] > arr[mid + 1]:
ans = mid
right = mid - 1
else:
left = mid + 1
return ans
我个人认为官方的解法跟我的思路是一样的,首先,我们假设峰顶的下标是i,那么i的左边永远是递增的,即在i的左边arr[j] < arr[j+1]恒成立,同理i的右边是递减的,在i的右边arr[j] > arr[j+1]恒成立。
并且在这个数组中只有峰顶i同时满足arr[i] > arr[i-1]和arr[i] > arr[i+1].
如果这两个条件不能同时满足,那就不是峰顶。
如果存在arr[i] < arr[i+1],说明峰顶在i 的右边,left 就往右移,即left = mid + 1.
如果存在arr[i] < arr[i-1],说明峰顶在i 的左边,right就往左移,即right = mid -1.