查看原文
其他

算法题 403:验证二叉搜索树

The following article is from 数据结构和算法 Author 山大王wld

(给算法爱好者加星标,修炼编程内功

来源: 数据结构和算法-山大王wld

问题描述


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

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

  • 节点的左子树只包含小于当前节点的数。

  • 节点的右子树只包含大于当前节点的数。

  • 所有左子树和右子树自身必须也是二叉搜索树。


示例 1:

输入:

    2

   / \

  1   3

输出: true

示例 2:

输入:

    5

   / \

  1   4

     / \

    3   6

输出: false

解释: 输入为: [5,1,4,null,null,3,6]。

     根节点的值为 5 ,但是其右子节点值为 4 。


递归写法


做这题之前我们首先要明白什么是二叉搜索树,就是每个节点左子树的值都比当前节点小,右子树的值都比当前节点大。所以看到这里我们最先想到的就是递归,我最先想到的是下面这种写法(注意是错误的)


1public boolean isValidBST(TreeNode root) {
2    if (root == null)
3        return true;
4    if (root.left != null && root.val <= root.left.val || root.right != null && root.val >= root.right.val)
5        return false;
6    return isValidBST(root.left) && isValidBST(root.right);
7}


如果一个结点是空的,我们默认他是有效的二叉搜索树,否则如果左节点不为空,我们要判断是否大于左节点的值,如果右节点不为空,我们还要判断小于右节点的值,然后我们再以左右两个子节点用相同的方式判断。看起来好像没什么问题,但我们好像忽略了一个每个节点的上限和下限,比如下面这棵树

注意6这个节点不光要小于15而且还要大于10,所以这里的每一个节点都是有一个范围的,上面的代码我只判断了6比15小,但没有和10进行比较,所以代码是错误的。这里我们来给每个节点添加一个范围,如果不在这个范围之内直接返回false,比如6的范围是(10,15),很明显他不在这个范围内,所以他不是二叉搜索树。根节点的范围我们从Long.MIN_VALUE到Long.MAX_VALUE,来看下代码


1public boolean isValidBST(TreeNode root) {
2    return isValidBST(root, Long.MIN_VALUE, Long.MAX_VALUE);
3}
4
5public boolean isValidBST(TreeNode root, long minVal, long maxVal) {
6    if (root == null)
7        return true;
8    //每个节点如果超过这个范围,直接返回false
9    if (root.val >= maxVal || root.val <= minVal)
10        return false;
11    //这里再分别以左右两个子节点分别判断,
12    //左子树范围的最小值是minVal,最大值是当前节点的值,也就是root的值,因为左子树的值要比当前节点小
13    //右子数范围的最大值是maxVal,最小值是当前节点的值,也就是root的值,因为右子树的值要比当前节点大
14    return isValidBST(root.left, minVal, root.val) && isValidBST(root.right, root.val, maxVal);
15}


中序遍历递归


根据二叉搜索树的性质我们知道,中序遍历二叉搜索树,遍历的结果一定是有序的。中序遍历时,判断当前节点是否大于中序遍历的前一个节点,也就是判断是否有序,如果不大于直接返回 false。


1//前一个结点,全局的
2TreeNode prev;
3
4public boolean isValidBST(TreeNode root) {
5    if (root == null)
6        return true;
7    //访问左子树
8    if (!isValidBST(root.left))
9        return false;
10    //访问当前节点:如果当前节点小于等于中序遍历的前一个节点直接返回false。
11    if (prev != null && prev.val >= root.val)
12        return false;
13    prev = root;
14    //访问右子树
15    if (!isValidBST(root.right))
16        return false;
17    return true;
18}


中序遍历非递归


如果对树的中序遍历比较熟悉的话,这里面也有树的中序遍历的递归和非递归两种写法。我们完全可以把上面中序遍历的递归改为非递归。


1public boolean isValidBST(TreeNode root) {
2    if (root == null)
3        return true;
4    Stack<TreeNode> stack = new Stack<>();
5    TreeNode pre = null;
6    while (root != null || !stack.isEmpty()) {
7        while (root != null) {
8            stack.push(root);
9            root = root.left;
10        }
11        root = stack.pop();
12        if (pre != null && root.val <= pre.val)
13            return false;
14        //保存前一个访问的结点
15        pre = root;
16        root = root.right;
17    }
18    return true;
19}


总结


这题可能最容易理解的是第一种解法,我们只需要给每个节点添加一个范围,然后再分别遍历每个节点,查看是否都在指定的范围内,只要有一个不在范围内,说明不是二叉搜索树,直接返回false。后面两种写法是根据二叉搜索树中序遍历的特点来判断,因为二叉搜索树中序遍历的结果是升序的,我们就按照二叉树中序遍历的方式来遍历这棵二叉树,然后在遍历的时候顺便保存一下前一个访问的结点,判断当前节点是否大于前一个结点的值,如果不大于直接返回false。



- EOF -


推荐阅读  点击标题可跳转

1、看了这篇文章,面试再也不怕树了

2、面试被问“红黑树”,我一脸懵逼......

3、算法题367:二叉树的最大深度


觉得本文有帮助?请分享给更多人

关注「算法爱好者」加星标,修炼编程内功

好文章,我在看❤️

    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存