首页 > 技术文章 > 【力扣】有效的山脉数组

Harukaze 2020-12-30 11:06 原文

来源:力扣(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行等号去掉,就会掠过值相同的情况从而造成程序错误。



 

 

 

 





推荐阅读