LeetCode669修剪二叉搜索树

题目链接

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

题解

  • 递归解法
  • 我看了题解写出来的,我分析题目的时候以为还要交换左右子树什么的……,其实不用
  • 思路见代码注释
// Problem: LeetCode 669
// URL: https://leetcode-cn.com/problems/trim-a-binary-search-tree/
// Tags: Tree BST Recursion
// Difficulty: Easy

struct TreeNode{
    int val;
    TreeNode* left;
    TreeNode* right;
};

class Solution{
public:
    TreeNode* trimBST(TreeNode* root, int L, int R) {
        // 处理空链表
        if (root == nullptr) return nullptr;
        // 如果根结点小于L,则根结点以及左子树会被剪掉,整个树的修剪结果是修剪后的右子树
        if (root->val < L) return trimBST(root->right, L, R);
        // 如果根结点小于L,则根结点以及右子树会被剪掉,整个树的修剪结果是修剪后的左子树
        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;
    }
};

作者:@臭咸鱼

转载请注明出处:https://www.cnblogs.com/chouxianyu/

欢迎讨论和交流!


posted @ 2020-08-01 10:40  臭咸鱼  阅读(132)  评论(0编辑  收藏  举报