LeetCode 669. Trim a Binary Search Tree

Posted 2021-01-01 約束の空

二叉查找树的套路一般都是 先根据 root->val 和一些条件缩小范围。

本题中，如果 根节点小于L，递归右子树，如果大于R，递归左子树。如果在两者之间，表明root一定是符合要求的，递归左子树和右子树，并和root接起来。

时间复杂度 O(n)，最多访问每个节点一次。

空间复杂度 O(h) 递归会调用栈，最多为树的高度。

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;
    }
};