给定一个二叉搜索树,同时给定最小边界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
注意到二叉树具有特征:对于任意一个节点,左子树上所有节点的值,都小于该节点的值,右子树的所有节点的值都大于该节点的值。因此在解题是可以利用该性质,当处理到某个非空节点时:
若该节点在[L, R]区间内,递归处理其左右子树;
若该节点的值大于R,则该节点的右子树的所有节点也大于R,可以全部舍去,用该节点的左子树代替该节点的位置,之后对于这个位置的新节点(原左子树根节点)递归处理;
若该节点的值小于L,则该节点的左子树的所有节点也小于L,可以全部舍去,用该节点的右子树替代该节点的位置,之后对于这个位置的新节点(原右子树根节点)递归处理。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode* trimBST(TreeNode* root, int L, int R) {
if (!root)
return nullptr;
if (root->val < L) {
return trimBST(root->right, L, R);
}
else if (root->val > R) {
return trimBST(root->left, L, R);
}
else {
root->left = trimBST(root->left, L, R);
root->right = trimBST(root->right, L, R);
return root;
}
}
};