给定一个二叉搜索树,同时给定最小边界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
解题思路:
递归。
- 若当前根结点为空,返回空。
- 如果当前的根结点值root->val<L,那么二叉搜索树的左子树所有值都小于L,所以剪去当前根结点,以及左子树。
- 如果root->val>R,那么二叉搜索树的右子树的所有值都大于R,所以应当剪去当前根结点,以及右子树。
- 若L<=root->val<=R,那么根结点无需修剪,递归修剪左右子树即可。
class Solution { public: TreeNode* trimBST(TreeNode* root, int L, int R) { if (!root) return NULL; if (root->val < L) return trimBST(root->right, L, R); if (root->val > R) return trimBST(root->left, L, R); root->left = trimBST(root->left, L, R); root->right = trimBST(root->right, L, R); return root; } }; |