题目描述:
标签:树 递归
给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构(即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在唯一的答案。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
代码:
思路分析:
1、递归结束条件,当遇到空结点返回null
2、如果root(当前节点)的元素小于low的数值,那么应该递归右子树,并返回右子树符合条件的头结点。(因为左子树肯定不满足条件,但右子树有可能有满足条件的节点,所以要递归找到该节点返回)
3、如果root(当前节点)的元素大于high的,那么应该递归左子树,并返回左子树符合条件的头结点。(因为右子树肯定不满足条件,但左子树有可能有满足条件的节点,所以要递归找到该节点返回)
/**
* Definition for a binary tree node.
* public 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 root;
if(root.val < low){
TreeNode right = trimBST(root.right,low,high);
return right;
}
if(root.val > high){
TreeNode left = trimBST(root.left,low,high);
return left;
}
root.left = trimBST(root.left,low,high);
root.right = trimBST(root.right,low,high);
return root;
}
}