LeetCode 669. Trim a Binary Search Tree

二叉查找树的套路一般都是 先根据 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;
    }
};

 

posted @ 2018-08-25 05:06  約束の空  阅读(78)  评论(0)    收藏  举报