判定一棵树是否满足二叉搜索树的性质。二叉查找树(Binary Search Tree),(又:二叉搜索树,二叉排序树)它或者是一棵空树,或者是具有下列性质的二叉树: 若它的左子树不空,则左子树上所有结点的值均小于它的根结点的值; 若它的右子树不空,则右子树上所有结点的值均大于它的根结点的值; 它的左、右子树也分别为二叉排序树。
下面是看到的一个大佬的思路讲解,非常清楚了,原文在:https://leetcode.com/problems/validate-binary-search-tree/discuss/158094/Python-or-Min-Max-tm Python | 给你把Min Max来由说透 - 公瑾™
> 类型:DFS遍历
> Time Complexity O(n)
> Space Complexity O(h)
LC里面各种花式炫技巧,炫答案,但没看到正儿八经给初学者解释关于Min Max运用的原理,这里公瑾给大家带来一个我最先犯的错误,然后是如何改进的。
class Solution(object):
def isValidBST(self, root):
return self.helper(root)
def helper(self, node):
if not node: return True
if node.left and node.left.val >= node.val:
return False
if node.right and node.right.val <= node.val:
return False
left = self.helper(node.left)
right = self.helper(node.right)
return left and right
上面代码看起来好像也没什么毛病,但以下这种情况是过不了的
5
/ \
1 4
/ \
3 6
为什么?因为我们每层在当前的root
分别做了两件事情:
root.left.val
是否比当前root.val
小root.right.val
是否比当前root.val
大大家可以用这个思路过一下上面这个例子,完全没问题。
那么问题来了,Binary Search Tree还有一个定义,就是
root.val
小root.val
大我们错就错在底层的3,比顶层的5,要小。
ok,概念弄懂了,如何解决这个问题呢?我们可以从顶层开始传递一个区间,举个例子。
在顶层5,向下传递的时候,
他向告诉左边一个信息:
左孩子,你和你的孩子,和你孩子的孩子,孩子的...........孩子都不能比我大哟
他向告诉右边一个信息:
右孩子,你和你的孩子,和你孩子的孩子,孩子的...........孩子都不能比我小哟
所以5告诉左边1的信息/区间是:(-infinite, 5)
所以5告诉右边4的信息/区间是:(5 , infinite)
然后我们要做的就是把这些信息带入到我们的代码里,我们把区间的左边取名lower_bound
, 右边取名upper_bound
这样才有了LC被复制到烂的标准
答案
class Solution(object):
def isValidBST(self, root):
return self.helper(root, -float('inf'), float('inf'))
def helper(self, node, lower_bound, upper_bound):
if not node: return True
if node.val >= upper_bound or node.val <= lower_bound:
return False
left = self.helper(node.left, lower_bound, node.val)
right = self.helper(node.right, node.val, upper_bound)
return left and right
这题呢,还有另外一个根据BST性质进行Inorder操作的答案
利用数组储存inorder过的数,如果出现重复,或者数组不等于sorted(arr)
,证明不是Valid Tree
这个解法比较易读,如果对Space Complexity要求不严格,可以通过比对数组里面的数而不是sorted(arr)
来达到O(N)时间复杂。
class Solution(object):
def isValidBST(self, root):
self.arr = []
self.inorder(root)
return self.arr == sorted(self.arr) and len(self.arr) == len(set(self.arr))
def inorder(self, root):
if not root: return
self.inorder(root.left)
self.arr.append(root.val)
self.inorder(root.right)
在上面的算法里进行了优化,每次只需要将当前root.val
和上次储存的self.last
比对即可知道是否满足条件。然后设立self.flag
用于返回。
class Solution(object):
def isValidBST(self, root):
self.last = -float('inf')
self.flag = True
self.inorder(root)
return self.flag
def inorder(self, root):
if not root: return
self.inorder(root.left)
if self.last >= root.val:
self.flag = False
self.last = root.val
self.inorder(root.right)