来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/valid-mountain-array
给定一个整数数组 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]
示例 1:
输入:[2,1] 输出:false
示例 2:
输入:[3,5,5] 输出:false
我在写这道题的代码的时候,首先使用暴力求解方法,思路很简单,先从小到大遍历数组,满足数值由小到大$idx$加一,再从大到小遍历数组,满足数值从大到小$ids$加一,当整个数组符合山脉数组的规律是$(idx+ids+1==len(arr))$一定成立,返回$True$值,但是我遇到一个问题,数组边界问题处理不好,第14行代码,如果$idx$与$j$都为0的话,$arr[0+0-1]=arr[-1]$,也就变成和数组最右边的元素进行比较了,破坏了比较顺序,必然产生Bug。
1 class Solution: 2 def validMountainArray(self,arr) -> bool: 3 idx = 0 4 ids = 0 5 if len(arr) >= 3: 6 if arr[1] <= arr[0]: 7 return False 8 for i, item in enumerate(arr): 9 if arr[i + 1] > arr[i]: 10 idx += 1 11 else: 12 break 13 for j, item in enumerate(arr[idx:]): 14 if arr[idx+j] < arr[idx+j-1]: 15 ids += 1 16 else: 17 break 18 19 if idx + ids + 1 == len(arr): 20 return True 21 else: 22 return False 23 24 else: 25 return False
经过修改与调试,代码被改成了这样
1 class Solution: 2 def validMountainArray(self,arr) -> bool: 3 idx = 1 4 ids = 0 5 if len(arr) >= 3: 6 if arr[1] <= arr[0]: 7 return False 8 for i in range(2,len(arr)): 9 if arr[i] > arr[i-1]: 10 idx += 1 11 if i == len(arr)-1: 12 return False 13 else: 14 break 15 for j in range(idx+1,len(arr)): 16 if arr[j] < arr[j-1]: 17 ids += 1 18 else: 19 break 20 21 if idx + ids + 1 == len(arr): 22 return True 23 else: 24 return False 25 26 else: 27 return False
从一开始使用$enumerate( )$迭代$i$与$j$就是不科学的,因为很难控制$i$,$j$的取值范围,这道题解这么久完全是因为$enumerate( )$方法的滥用导致的,换成range( ),明显感觉到$i$,$j$的范围好控制多了,至于边界问题的处理,按照从左到右的思考逻辑为了避免$arr[i] > arr[i-1]$出现 $arr[1] > arr[-1]$的比较情况,我们从一开始就设定了边界使i的取值从2开始,至于$arr[1]$与$arr[0]$的比较情况我们在进入循环体之前另行判断。
占用的内存空间非常惊人
最后附上大佬的思路比较学习:
这个题我们用举旗子法可以很好的解决。
因为题目说了是“山”,山必须只能有一个峰。
也就是说数组一定是要先增后降的。
先来两个基础判断。如果$len(A) < =2$或$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$。
结束啦。
1 class Solution: 2 def validMountainArray(self, A: List[int]) -> bool: 3 if len(A) <= 2: #谢谢抓克大佬修正 4 return False 5 6 if A[0] >= A[1]: 7 return False 8 9 flag = 1 10 for i in range(2, len(A)): 11 if flag and A[i - 1] >= A[i]: # 发现要下山啦 12 flag = 0 13 if not flag and A[i - 1] <= A[i]: # 又要上山啦 14 return False 15 16 if flag == 1: # 一直在上山 17 return False 18 else: 19 return True 20 21 22 作者:flying_du 23 链接:https://leetcode-cn.com/problems/valid-mountain-array/solution/python-ju-qi-zi-fa-quan-guo-zui-cai-xiao-qi-zi-by-/ 24 来源:力扣(LeetCode)
可以看到为了避免出现数组边界问题,$i$取值也是从2开始的。
至于为什么11行与13行要加等号可能有些费解我来走一遍循环体可能就清晰了:
输入$[6,7,7,8,6]$,预期结果$False$
$i=2$, 满足语句$if 1 and A[1] >= A[2]: flag=0$,紧接着又满足$if 1 and A[1] <= A[2]$,循环结束
连续两个$if$起到了判断相同值的作用,如果把11行等号去掉,就会掠过值相同的情况从而造成程序错误。