给定一个整数数组 arr
,如果它是有效的山脉数组就返回 true
,否则返回 false
。
让我们回顾一下,如果 arr
满足下述条件,那么它是一个山脉数组:
arr.length >= 3
- 在
0 < i < arr.length - 1
条件下,存在i
使得:arr[0] < arr[1] < ... arr[i-1] < arr[i]
arr[i] > arr[i+1] > ... > arr[arr.length - 1]
示例 1:
输入:arr = [2,1] 输出:false
示例 2:
输入:arr = [3,5,5] 输出:false
示例 3:
输入:arr = [0,3,2,1] 输出:true
思路:首先山脉数组长度得大于3,然后左半部分严格递增,后半部分严格递减。
可能有些人开始会用index = len(arr)/2,然后左半部分检查是否严格递增,右半部分是否严格递减,这就跟我一样陷入想当然的怪圈了,它只说是峰顶,但不一样刚好在中间位置!
比如数组[1,2,3,4,7,5],峰顶就在右边倒数第二个位置。
因此,该怎么做呢,峰顶元素一定是整个数组的最大值,因此我们先找到最大值所在的位置,如果出现在了数组的最左边或者最右边,也就是不形成一个山脉了,那肯定直接返回False,这一次遍历寻找到峰顶元素,如果没返回false,也就保证了左半边一定是严格递增的,那我们接下去只需要遍历右半边是否严格递减即可。
代码(Python):
class Solution(object):
def validMountainArray(self, arr):
if len(arr) < 3:
return False
# 找到峰顶的位置
peak = 0
for i in range(1, len(arr)):
if arr[i] > arr[i-1]:
peak = i
else:
break
# 如果没有上升的部分或峰顶位于数组的开头或结尾,则不是山脉数组
if peak == 0 or peak == len(arr) - 1:
return False
# 检查从峰顶开始是否是严格下降的
for i in range(peak, len(arr)-1):
if arr[i] <= arr[i+1]:
return False
return True