【二叉搜索树】LeetCode 98. 验证二叉搜索树【中等】

给你一个二叉树的根节点 root ,判断其是否是一个有效的二叉搜索树。

有效 二叉搜索树定义如下:

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

示例 1:

 

 

输入:root = [2,1,3]
输出:true

示例 2:

输入:root = [5,1,4,null,null,3,6]
输出:false
解释:根节点的值是 5,但是右子节点的值是 4。

分析

方法一:递归

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

这启示我们设计一个递归函数helper(root, lower, upper)来递归判断,函数表示考虑以root为根的子树,判断子树中所有节点的值是否都在(l, r)的范围内(注意是开区间)。若root节点的值val不在(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表示一个无穷大的值。

下图展示了算法如何在示例2上运行:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isValidBST(self, root: Optional[TreeNode]) -> bool:
        def BFS(root, lower, upper):
            if not root: return True

            if lower < root.val < upper:
                return BFS(root.left, lower, root.val) and BFS(root.right, root.val, upper)
            else:
                return False

        return BFS(root, -float('inf'), float('inf'))

时间复杂度:O(n),其中 n 为二叉树的节点个数。在递归调用的时候二叉树的每个节点最多被访问一次,因此时间复杂度为 O(n)。

空间复杂度:O(n),其中 n 为二叉树的节点个数。递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,即二叉树的高度。最坏情况下二叉树为一条链,树的高度为 n ,递归最深达到 n 层,故最坏情况下空间复杂度为 O(n) 。

 

方法二:中序遍历

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

中序遍历是二叉树的一种遍历方式,它先遍历左子树,再遍历根节点,最后遍历右子树。而我们二叉搜索树保证了左子树的节点的值均小于根节点的值,根节点的值均小于右子树的值,因此中序遍历以后得到的序列一定是升序序列。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isValidBST(self, root: Optional[TreeNode]) -> bool:
        stack, inorder = [], float('-inf')
        while stack or root: # 当栈或者根节点非空
            while root: # 当根节点非空
                stack.append(root) # 将根节点入栈
                root = root.left # 找到根节点的左子树,将其设置为新的root
            root = stack.pop() # 直到左子树为空,将最近一个root也就是左节点出栈

            # 若中序遍历得到的节点的值小于等于前一个inorder, 说明不是二叉搜索树
            if root.val <= inorder: 
                return False 
            inorder = root.val
            root = root.right # 继续遍历右子树
        return True

# 时间复杂度:O(n),其中 n 为二叉树的节点个数。二叉树的每个节点最多被访问一次,因此时间复杂度为 O(n)。
# 空间复杂度:O(n),其中 n 为二叉树的节点个数。栈最多存储 n 个节点,因此需要额外的 O(n) 的空间。

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 


 

提示:

树中节点数目范围在[1, 104] 内
-231 <= Node.val <= 231 - 1

 

posted @ 2022-05-27 20:32  Ariel_一只猫的旅行  阅读(26)  评论(0编辑  收藏  举报