題目描述

給定一個(gè)二叉搜索樹(shù)，同時(shí)給定最小邊界L 和最大邊界 R。通過(guò)修剪二叉搜索樹(shù)，使得所有節(jié)點(diǎn)的值在[L, R]中 (R>=L) 。你可能需要改變樹(shù)的根節(jié)點(diǎn)，所以結(jié)果應(yīng)當(dāng)返回修剪好的二叉搜索樹(shù)的新的根節(jié)點(diǎn)。

示例 1:

輸入: 
    1
   / \
  0   2

  L = 1
  R = 2

輸出: 
    1
      \
       2

示例 2:

輸入: 
    3
   / \
  0   4
   \
    2
   /
  1

  L = 1
  R = 3

輸出: 
      3
     / 
   2   
  /
 1

Solution

遞歸

public class TreeNode {

    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

public static 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.right = trimBST(root.right,L,R);
        root.left = trimBST(root.left,L,R);

        return root;
    }

解題思路：

從根節(jié)點(diǎn)開(kāi)始判斷大小，比L小或者比R大的值就剪去

比R大，返回左節(jié)點(diǎn)遞歸后的節(jié)點(diǎn);比L小，返回右節(jié)點(diǎn)遞歸后的值

在L,R之間的話(huà)，遞歸的判斷左右節(jié)點(diǎn)的值

LettCode-修剪二叉搜索樹(shù)

GitHub

微信公眾號(hào):一只愛(ài)生活的程序猿