(Java)leetcode-669 Trim a Binary Search Tree(修剪二叉搜索树)

题目描述

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

在这里插入图片描述
在这里插入图片描述

思路

修剪BST树,其步骤为:
若为空树,返回NULL;
否则:
1.不正常的节点,先修剪之,若根的值不在[L,R]范围内,则执行如下修剪:
若根小于下限L,必然有其左子树结点全部小于L(二叉搜索树的性质),放弃根和左子树,使右子树的根成为新树的根。这里的做法是递归调用 trimBST() 方法,入参为右子树的根,那么右子树也会进行这一步的调整,因为右子树还可能依然不符合要求…最终返回的是符合要求的子树
大于上限R的情况同理。最后结束时,要么根为空,要么根的值在[L,R]中。
2.正常的节点,递归修剪其孩子

代码

class Solution {
    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;
    }
}

错误代码

class Solution {
    public TreeNode trimBST(TreeNode root, int L, int R) {
    	check(root, L, R);
    	return root;
    }

    private void check(TreeNode root, int L, int R) {
    	if (root == null) return;

		while (root  && root.val < L )  root = root.right; // 这里只是修改了临时变量(root)所指向的对象,而原树的结构没有改变
		while (root  && root.val > R)  root = root.left;
		
		if (root != null) {
			check(root.left, L, R); 
			check(root.right, L, R); 
		}
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值