[LeetCode] Trim a Binary Search Tree

Posted 2020-10-06 immjc

Given a binary search tree and the lowest and highest boundaries as L and R, trim the tree so that all its elements lies in [L, R] (R >= L). You might need to change the root of the tree, so the result should return the new root of the trimmed binary search tree.

Example 1:

Input: 
    1
   /   0   2

  L = 1
  R = 2

Output: 
    1
             2

Example 2:

Input: 
    3
   /   0   4
       2
   /
  1

  L = 1
  R = 3

Output: 
      3
     / 
   2   
  /
 1

修剪一个二叉树：给定一个二叉树的最大和最小边界L和R，使树中的元素位于L和R之间。

1. 当root位于L和R之间时，递归的修剪其左右子树，并返回root。

2. 当root的值小于L，其左子树都小于L，故舍弃其左子树，递归的修剪其右子树，并返回修剪过的右子树。

3. 当root的值大于R，其右子树都大于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) {
        if (root == nullptr)
            return 0;
        if (root->val < L)
            return trimBST(root->right, L, R);
        else if (root->val > R) 
            return trimBST(root->left, L, R);
        else {
            root->left = trimBST(root->left, L, R);
            root->right = trimBST(root->right, L, R);
            return root;
        }
    }
};
// 16ms