每日力扣-树-修剪二叉搜索树

124 阅读1分钟

给你二叉搜索树的根节点 root ,同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树,使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构(即,如果没有被移除,原有的父代子代关系都应当保留)。 可以证明,存在唯一的答案。

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意,根节点可能会根据给定的边界发生改变。

解题思路:因为是二叉搜索树,如果当前节点比左边界小,那他的左子树肯定都比左边界小,全部一起舍弃,所以返回右子树。同理大边一样。

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if (root == NULL) 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;
    }
};