算法学习：669. 修剪二叉搜索树

修剪二叉搜索树

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

示例

输入：root = [1,0,2], low = 1, high = 2
输出：[1,null,2]

思路

如果一个节点的值没有落在 [lo, hi] 中，有两种情况：
1、root.val < lo，这种情况下 root 节点本身和 root 的左子树全都是小于 lo 的，都需要被剪掉。

2、root.val > hi，这种情况下 root 节点本身和 root 的右子树全都是大于 hi 的，都需要被剪掉

二叉树节点的定义

 //二叉树节点的定义。
 class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(){}
     TreeNode(int val){
         this.val = val;
     }
     TreeNode(int val,TreeNode left,TreeNode right){
         this.val = val;
         this.left = left;
         this.right = right;
     }
}

递归代码

class Solution{
	public TreeNode trimBST(TreeNode root, int low, int high) {
		if(root == null)	return null;
		if(root.val < low)	
			// 直接返回 root.right
            // 等于删除 root 以及 root 的左子树
			return trimBST(root.right,low,high);
		if(root.val > high)	
			// 直接返回 root.left
            // 等于删除 root 以及 root 的右子树
			return trimBST(root.left,low,high);
		// 闭区间 [lo, hi] 内的节点什么都不做
		root.left = trimBST(root.left,low,high);
		root.right = trimBST(root.right,low,high);
		return root;
    }
}