LeetCode | 669.修剪二叉搜索树

LeetCode | 669.修剪二叉搜索树

给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。

解答

• 首先要知道二叉搜索树的性质：根的值是大于左结点的值，小于右结点的值。
• 利用以上性质，可做以下的递归思想：
• 如果当前节点的值是小于L的值，则说明包含该节点在内和左子树中所有节点的值都是小于L，因此可直接舍弃包含该节点在内的左子树。然后继续往右子树走，直到找到>=L的节点。
• 同样，如果当前节点的值是大于R的值，则说明包含该节点在内和右子树中所有节点的值都是大于R的，因此可直接舍弃包含该系欸但在内的右子树。然后继续往左子树走，直到找到<=R的节点。
• 最后，如果当前节点是在[L, R]的范围内的，则说明当前节点是符合的，则我们就要找到适合当前节点的左孩子还右孩子，所以要继续分别往左边和右边走。

/**
 * 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) {
        // TODO:
        // 如何释放舍弃节点的内存
        if (!root)
            return root;
        // 如果当前节点值小于L，则说明该节点的左子树都小于L
        // 所以可直接舍弃该节点的左子树，然后再继续往右子树
        // 找到>=L的节点
        if (root->val < L){
            // 可以再这里调用一个内存释放函数，释放包括root节点在内的所有左子树的节点
            // 不过释放在在先获得root->right的内存地址，防止root被释放后，而使用不了。
            return trimBST(root->right, L, R);
        }
            
        // 如果当前节点值大于R，则说明该节点的右子树都大于R
        // 所以可直接舍弃该节点的右子树，然后再继续往左子树
        // 找打<=R的节点
        else if (root->val > R){
            // 可以再这里调用一个内存释放函数，释放包括root节点在内的所有右子树子树的节点
            // 不过释放在在先获得root->left的内存地址，防止root被释放后，而使用不了。
            return trimBST(root->left, L, R);
        }
            
        // 如果以上两者情况都不满足，则说明该节点是满足的
        // 可继续遍历下去继续修剪
        else{
            // 首先我们先找到合适当前节点的左孩子节点，往左边走
            root->left = trimBST(root->left, L, R);
            // 其次再找到合适当前节点的右孩子节点，往右边走
            root->right = trimBST(root->right, L, R);
            return root;
        }
    }
};