LeetCode669. 修剪二叉搜索树

最新推荐文章于 2022-10-17 15:31:42 发布

wjgxajh

最新推荐文章于 2022-10-17 15:31:42 发布

阅读量141

点赞数

分类专栏： Leetcode经典算法题目分析文章标签：二叉树

本文链接：https://blog.csdn.net/wjgxajh/article/details/84101066

版权

Leetcode经典算法题目分析专栏收录该内容

27 篇文章 0 订阅

订阅专栏

给定一个二叉搜索树，同时给定最小边界L 和最大边界 R。通过修剪二叉搜索树，使得所有节点的值在[L, R]中 (R>=L) 。你可能需要改变树的根节点，所以结果应当返回修剪好的二叉搜索树的新的根节点。
示例 1:

输入:
1
/
0 2

L = 1
R = 2

输出:
1

2
示例 2:

输入:
3
/
0 4

2
/
1

L = 1
R = 3

输出:
3
/
2
/
1

思路：分治法，分别修剪左子树和右子树

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
static const int _ = []() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    return 0;
}();
class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        if (root == NULL)
			return NULL;
        if(root->val < L) // 根节点小于等于最小值
            return this->trimBST(root->right, L, R);
        else if (root->val > R)// 根节点大于等于最大值
            return this->trimBST(root->left, L, R);
        else
        {
            // 先比较左子树
			root->left = this->trimBST(root->left, L, R);
			// 再比较右子树
			root->right = this->trimBST(root->right, L, R);
        }
		return root;
    }
};