给定一个整数数组 A
,如果它是有效的山脉数组就返回 true
,否则返回 false
。
让我们回顾一下,如果 A 满足下述条件,那么它是一个山脉数组:
A.length >= 3
- 在
0 < i < A.length - 1
条件下,存在i
使得:A[0] < A[1] < ... A[i-1] < A[i]
A[i] > A[i+1] > ... > A[B.length - 1]
示例 1:
输入:[2,1] 输出:false
示例 2:
输入:[3,5,5] 输出:false
示例 3:
输入:[0,3,2,1] 输出:true
提示:
0 <= A.length <= 10000
0 <= A[i] <= 10000
思路:
题目要求如下:
1. 长度 > 3
2. 可以分成两段,第一段严格递增,第二段严格递减。(严格的意思是不能出现重复)
3. 最高点不能是第一个点或者最后一个点。
找到两段之后,用转集合比较长度的方法判断有没有重复元素,
再用排序的方法判断是不是严格递增和严格递减。
class Solution(object):
def validMountainArray(self, A):
"""
:type A: List[int]
:rtype: bool
"""
if len(A) < 3:
return False
max_A_pos = A.index(max(A))
if max_A_pos in [0, len(A) - 1]:
return False
first, last = A[:max_A_pos + 1], A[max_A_pos:]
if len(first) != len(set(first)) or len(last) != len(set(last)): #判断有没有重复元素
return False
return first == sorted(first) and last == sorted(last)[::-1]