LeetCode

修剪二叉搜索树

2019-07-08  本文已影响0人  习惯了_就好

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

示例 1:

输入:
1
/
0 2

L = 1
R = 2

输出:
1

2

示例 2:

输入:
3
/
0 4

2
/
1

L = 1
R = 3

输出:
3
/
2
/
1

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trim-a-binary-search-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //二叉搜索树,根节点大于左子树节点的值,小于右子树节点的值
    public TreeNode trimBST(TreeNode root, int L, int R) {
        if(root == null){
            return root;
        }
        
        
        if(root.val < L){//根节点比最小值小,只能去右子树里面取值
            return trimBST(root.right, L, R);
        } else if(root.val > R){//根节点比最大值大,只能去左子树里面取值
            return trimBST(root.left, L, R);
        } else {//根节点在两者之间,需要裁减左子树和右子树
            root.left = trimBST(root.left, L, R);
            root.right = trimBST(root.right, L, R);
            return root;
        }
        
    }
}
上一篇下一篇

猜你喜欢

热点阅读