题目描述:
给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构(即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在唯一的答案。
所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。
链接:https://leetcode-cn.com/problems/trim-a-binary-search-tree
输入: root = [3,0,4,null,2,null,null,1], low = 1, high = 3
输出:[3,2,null,1]
思路分析:二叉搜索树中,其中的值必须在当前区间中[low,high]
。
- 节点值在该区间,则返回该节点
- 节点值不在该区间,返回左孩子或者右孩子
- 递归上述情况
代码:
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int low, int high) {
if(root == nullptr)//为空直接返回
return nullptr;
root->left = trimBST(root->left,low,high);//左子树
root->right = trimBST(root->right,low,high);//右子树
if(root->val < low || root->val > high)//当前值不在该区间,返回左孩子或者右孩子
{
if(root->left)
return root->left;
if(root->right)
return root->right;
return nullptr;
}
return root;//满足情况,返回该节点
}
};