給定一個二叉搜索樹,同時給定最小邊界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
/**
* 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 null;
}else if(root.val > R){
return trimBST(root.left,L,R);
}else 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;
}
}