Skip to content

验证二叉搜索树

约 912 字大约 3 分钟

DFS前序中序后序

2025-02-25

98. 验证二叉搜索树

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

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

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

示例 1:

img.png

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

示例 2:

img.png

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

提示:

  • 树中节点数目范围在[1, 104]
  • -2^31 <= Node.val <= 2^31 - 1

前序遍历,归并结果

左子树的上界是根节点的值,下界是最小值。右子树的上界是最大值,下界是根节点的值。

public class Solution {
    // 主函数,检查树是否为有效的二叉搜索树
    public boolean isValidBST(TreeNode root) {
        // 调用辅助函数,初始范围设为 [Long.MIN_VALUE, Long.MAX_VALUE]
        // 因为节点值可能取到Integer的最小值或最大值,因此范围应该是Long的最小值和最大值。
        return isValidBST(root, Long.MIN_VALUE, Long.MAX_VALUE);
    }

    // 辅助函数,递归检查树是否为有效的二叉搜索树
    public boolean isValidBST(TreeNode node, long lower, long upper) {
        // 递归结束条件。如果当前节点为空,返回 true(空树是有效的二叉搜索树)
        if (node == null) {
            return true;
        }
        // 特殊条件。如果当前节点的值不在合法范围内,返回 false
        if (node.val <= lower || node.val >= upper) {
            return false;
        }
        
        // 递归检查左子树和右子树
        // 对左子树,当前节点值是新的上界
        // 对右子树,当前节点值是新的下界
        return isValidBST(node.left, lower, node.val) && isValidBST(node.right, node.val, upper);
    }
}
  • 时间复杂度: O(n)O(n),其中 nn 为二叉树的节点个数。在递归调用的时候二叉树的每个节点最多被访问一次,因此时间复杂度为 O(n)O(n)
  • 空间复杂度: O(n)O(n),其中 nn 为二叉树的节点个数。递归函数在递归过程中需要为每一层递归函数分配栈空间,所以这里需要额外的空间且该空间取决于递归的深度,即二叉树的高度。最坏情况下二叉树为一条链,树的高度为 nn ,递归最深达到 nn 层,故最坏情况下空间复杂度为 O(n)O(n)

中序遍历

根据递归写迭代形式的解法。

结束条件,节点为空,返回true。

先递归遍历左子树,再检查当前节点值是否大于前一个中序遍历的节点值,并更新更新中序遍历的前一个节点值为当前节点值。最后递归遍历右子树。

class Solution {
    // 为什么要用-Long.MAX_VALUE?因为一开始比较的时候 node.val 必须大于 -Integer.MAX_VALUE
    private long inorder = -Long.MAX_VALUE;

    public boolean isValidBST(TreeNode root) {
        return inorderTraversal(root);
    }

    private boolean inorderTraversal(TreeNode node) {
        if (node == null) {
            return true;
        }

        // 递归遍历左子树
        if (!inorderTraversal(node.left)) {
            return false;
        }
        // 检查当前节点值是否大于前一个中序遍历的节点值。此时inorder的值是左子树的最右下的节点的值。
        if (node.val <= inorder) {
            return false;
        }
        // 更新inorder为当前节点值,以方便下次递归时与中序遍历的下一个节点比较。重点。
        inorder = node.val;
        // 递归遍历右子树
        return inorderTraversal(node.right);
    }
}

根据递归可以写迭代形式的中序遍历,即把栈描述出来。