669. Trim a Binary Search Tree

最新推荐文章于 2025-07-28 20:35:49 发布

linshuqun4

最新推荐文章于 2025-07-28 20:35:49 发布

阅读量148

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode 算法分析与设计

本文链接：https://blog.youkuaiyun.com/linshuqun4/article/details/78517194

本文介绍了一种使用递归方法重构二叉树的技术，确保所有节点的值在一个指定的范围内。文章详细解释了递归的过程，并提供了实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目的要求是，给定一个二叉树跟一个数值区间，现在需要我们重构二叉树，使得每个节点的值都在区间内里面。例子如下：

Example 1:

Input: 
    1
   / \
  0   2

  L = 1
  R = 2

Output: 
    1
      \
       2

面对这道题，我们最直观的想法是用递归去解决，因为根据题目的要求，我们对每一个节点的操作都是一样的：判断节点值在不在区间内，从而决定该节点要不要保留。而一个节点的判断情况有三种，令节点值为value：

1.节点值在L<=value<=R，保留节点，判断节点的左子树跟右子树

2.节点值value<L，舍弃节点，同时，因为节点的左子树的值全都小于value，所以左子树也舍弃，直接返回右子树的判断结果

3.节点值value>R，舍弃节点，同时，因为节点的右子树的值全都大于value，所以右子树也舍弃，直接返回左子树的判断结果

至此，一个节点的判断过程结束，我们也可是实现上述的递归过程。

完整的代码如下：

/**
 * 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 == NULL) return root;
        if (root->val < L) {
            return trimBST(root->right, L, R);
        } else 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;
    }
};