【Leetcode】669. Trim a Binary Search Tree

题目地址：

https://leetcode.com/problems/trim-a-binary-search-tree/

给定一棵二叉搜索树，再给定一个范围$[L,R]$，要求将这棵树所有在这个范围之外的节点都删掉，返回新树树根。

思路是用分治法。如果是空树则直接返回，否则看树根的范围，如果树根处于$(-\infty ,L)$，那说明树根以及整个左子树都要被砍掉，所以只需返回trim(root.right, L, R)；如果树根处于$(R,+\infty )$，那说明树根以及整个右子树都要被砍掉，所以只需返回trim(root.left, L, R)；否则的话，说明树根不需要被砍掉，只需分而治之，把左右子树trim完的树根返回过去与原树根连上即可。代码如下：

class Solution {
 public:
  TreeNode* trimBST(TreeNode* root, int l, int r) {
    if (!root) return nullptr;
    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;
  }
};

时间复杂度$O(n)$，空间$O(h)$。