LeetCode 669. Trim a Binary Search Tree

最新推荐文章于 2025-02-03 15:37:10 发布

原创最新推荐文章于 2025-02-03 15:37:10 发布 · 232 阅读

0 ·

CC 4.0 BY-SA版权

C/C++ 同时被 3 个专栏收录

265 篇文章

订阅专栏

LeetCode

125 篇文章

订阅专栏

Tree

17 篇文章

订阅专栏

本文介绍了一种算法，用于修剪二叉搜索树，使其所有元素位于指定的最低和最高边界[L, R]之间。通过递归地修剪不符合条件的节点，确保最终树的所有元素都符合给定范围。

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

题目：
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:

Output:

1
  \
   2

Example 2:
Input:

Output:

思路：如果节点为空，直接返回NULL；如果节点不为空，如果当前节点的值小于L，递归右子节点；如果当前节点的值大于R，递归左子节点；如果介于L、R之间，那么root的左右子节点分别递归。

代码：

/**
 * 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){//如果节点不为空
            if(root->val<L){//如果当前节点的值小于L，递归右子节点
                return trimBST(root->right,L,R);
            }
            if(root->val>R){//如果当前节点的值大于R，递归左子节点
                return trimBST(root->left,L,R);
            }
            root->left=trimBST(root->left,L,R);//如果介于L、R之间，那么root的左右子节点分别递归
            root->right=trimBST(root->right,L,R);
            return root;
        }
        else{//如果节点为空，直接返回NULL
            return NULL;
        }
    }
};

结果： 18ms