- 题目:
给定一个二叉搜索树,同时给定最小边界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
- 解题思路
若节点值小于最小边界值,减掉左子树;若节点值大于最大边界值,剪掉右子树;否则,戏修剪节点的左右子树
代码实现(c++)
TreeNode* trimBST(TreeNode* root, int L, int R) {
if(root == nullptr)
return root;
if(root->val < L) //剪去左子树
return trimBST(root->right, L, R);
if(root->val > R) //剪去右子树
return trimBST(root->left, L, R);
root->right = trimBST(root->right, L, R); //修剪左子树
root->left = trimBST(root->left, L, R); //修剪右子树
return root;
}