题目

给定一个二叉树,判断其是否是一个有效的二叉搜索树。

假设一个二叉搜索树具有如下特征:

  • 节点的左子树只包含小于当前节点的数。
  • 节点的右子树只包含大于当前节点的数。
  • 所有左子树和右子树自身必须也是二叉搜索树。

image.png

思路

方法一: 递归

思路和算法

要解决这道题首先我们要了解二叉搜索树有什么性质可以给我们利用,由题目给出的信息我们可以知道:如果该二叉树的左子树不为空,则左子树上所有节点的值均小于它的根节点的值; 若它的右子树不空,则右子树上所有节点的值均大于它的根节点的值;它的左右子树也为二叉搜索树。

这启示我们设计一个递归函数 helper(root, lower, upper) 来递归判断,函数表示考虑以 root 为根的子树,判断子树中所有节点的值是否都在 (l,r)(l,r) 的范围内(注意是开区间)。如果 root 节点的值 val 不在 (l,r)(l,r) 的范围内说明不满足条件直接返回,否则我们要继续递归调用检查它的左右子树是否满足,如果都满足才说明这是一棵二叉搜索树。

那么根据二叉搜索树的性质,在递归调用左子树时,我们需要把上界 upper 改为 root.val,即调用 helper(root.left, lower, root.val),因为左子树里所有节点的值均小于它的根节点的值。同理递归调用右子树时,我们需要把下界 lower 改为 root.val,即调用 helper(root.right, root.val, upper)。

函数递归调用的入口为 helper(root, -inf, +inf), inf 表示一个无穷大的值。

  1. # Definition for a binary tree node.
  2. # class TreeNode:
  3. # def __init__(self, x):
  4. # self.val = x
  5. # self.left = None
  6. # self.right = None
  7. class Solution:
  8. def isValidBST(self, root: TreeNode) -> bool:
  9. # 递归
  10. def helper(root, left=-float('INF'), right=float('INF')):
  11. if root is None:
  12. return True
  13. if not left < root.val < right:
  14. return False
  15. # 当前节点满足递归二叉树的条件
  16. # 判断当前节点的左右节点是否满足
  17. return helper(root.left, left, root.val) and helper(root.right, root.val, right)
  18. return helper(root)

方法二:中序遍历

思路和算法

基于方法一中提及的性质,我们可以进一步知道二叉搜索树「中序遍历」得到的值构成的序列一定是升序的,这启示我们在中序遍历的时候实时检查当前节点的值是否大于前一个中序遍历到的节点的值即可。如果均大于说明这个序列是升序的,整棵树是二叉搜索树,否则不是,下面的代码我们使用栈来模拟中序遍历的过程。

可能由读者不知道中序遍历是什么,我们这里简单提及一下,中序遍历是二叉树的一种遍历方式,它先遍历左子树,再遍历根节点,最后遍历右子树。而我们二叉搜索树保证了左子树的节点的值均小于根节点的值,根节点的值均小于右子树的值,因此中序遍历以后得到的序列一定是升序序列。

  1. class Solution:
  2. def isValidBST(self, root):
  3. """
  4. :type root: TreeNode
  5. :rtype: bool
  6. """
  7. stack, inorder = [], float('-inf')
  8. while stack or root:
  9. while root:
  10. stack.append(root)
  11. root = root.left
  12. root = stack.pop()
  13. # 如果中序遍历得到的节点的值小于等于前一个 inorder,说明不是二叉搜索树
  14. if root.val <= inorder:
  15. return False
  16. inorder = root.val
  17. root = root.right
  18. return True