首页 > 其他 > 详细

LeetCode——修剪二叉搜索树

时间:2020-05-18 13:19:51      阅读:52      评论:0      收藏:0      [点我收藏+]

Q:给定一个二叉搜索树,同时给定最小边界L?和最大边界?R。通过修剪二叉搜索树,使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点,所以结果应当返回修剪好的二叉搜索树的新的根节点。
技术分享图片
技术分享图片

A:
思路分析:

  • 修剪一棵树,如果根结点的值小于给定的左边界L,那么当前结点及其左子树就会被修剪掉,修剪后的树应该是其右子树,但是右子树不一定是符合范围的树,所以要对其右子树叶进行修剪,然后返回修剪后的右子树。
  • 同理,根结点的值大于给定的右边界R,修剪后的树应该是其左子树且要对左子树修剪。
  • 涉及到改变树的结构,就需要更新链接,如果当前结点值在范围内,那么修建其左右子树,并且更新左右链接。最后将当前修剪好的子树返回。

时间复杂度为O(n), 空间复杂度与树高成正比。

代码:

    public TreeNode trimBST(TreeNode root, int L, int R) {
        if (root == null)
            return null;
        if (root.val < L)
            return trimBST(root.right, L, R);
        if (root.val > R)
            return trimBST(root.left, L, R);
        root.left = trimBST(root.left, L, R);
        root.right = trimBST(root.right, L, R);
        return root;
    }

LeetCode——修剪二叉搜索树

原文:https://www.cnblogs.com/xym4869/p/12909842.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!