搜索
您的当前位置:首页修剪二叉搜索树

修剪二叉搜索树

来源:乌哈旅游

给定一个二叉搜索树,同时给定最小边界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

思路:递归。

如果当前val > R ,那就减掉右边,也就是修剪后的二叉树是左边的;

如果当前val < L ,那就减掉左边,也就是修剪后的二叉树是右边的;

如果当前的val 在[L, R] 这个范围,那么递归检查其左子树与右子树,即如果发现左右子树有不合法的,也要进行修剪。

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

        // 当前值合法,就要检查其子树是否合法
        root.left = trimBST(root.left,L,R);
        root.right = trimBST(root.right,L,R);
        return root;
    }

 

因篇幅问题不能全部显示,请点此查看更多更全内容

Top