【笔记】LeetCode 941. 有效的山脉数组

941. 有效的山脉数组

难度 简单

---

给定一个整数数组 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[A.length - 1]

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-fU4IgLRk-1604392451805)(hint_valid_mountain_array.png)]

示例 1：

输入：[2,1]
输出：false

示例 2：

输入：[3,5,5]
输出：false

示例 3：

输入：[0,3,2,1]
输出：true

提示：

0 <= A.length <= 10000
0 <= A[i] <= 10000 

---

菜狗子的代码

class Solution(object):
    def validMountainArray(self, A):
        # A.length>=3
        if len(A) < 3:
            return False
        # temp 用作比较的数， flag记录上升（1）或下降（0）
        temp = A[0]
        flag = 1
        for i in range(1, len(A)):
            # 升
            if flag:
                if A[i] < temp:
                    flag = 0
                    if A[0] == temp:
                        return False
                elif A[i] == temp:
                    return False
            # 降
            else:
                if A[i] >= temp:
                    return False
            temp = A[i]
        if flag:
            return False
        else:
            return True
                    
if __name__ == "__main__":
    A = [9,8,7]
    so = Solution()
    print(so.validMountainArray(A))

大神思路

用举旗子法可以很好的解决。

因为题目说了是“山”，山必须只能有一个峰。
也就是说数组一定是要先增后降的。

先来两个基础判断。如果len(A) < 3或A[0] >= A[1]就直接返回False。
然后就拿出我们的小旗子。正常情况下肯定是在上山的，我们flag = 1，当正在上山，发现A[i - 1] >= A[i]开始下山了，我们就flag = 0。
如果flag = 0的时候又发现上山了（A[i - 1] <= A[i]），就直接返回False。

这里注意一下，如果我们的小旗子一直没变。一直flag = 1，说明没有下过山，那也是不对的。也要返回False。

按照大神思路写的代码

class Solution(object):
    def validMountainArray(self, A):
        if len(A) < 3 or A[0] >= A[1]:
            return False
        flag = 1
        for i in range(1, len(A)):
            if flag:
                if  A[i - 1] > A[i]:
                    flag = 0
            else:
                if A[i - 1] <= A[i]:
                    return False
        return flag == 0