Java C++ 算法题解leetcode669修剪二叉搜索树示例

更新时间：2022年09月14日 09:43:39 作者：AnjaVon

这篇文章主要为大家介绍了Java C++ 算法题解leetcode669修剪二叉搜索树示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪

题目要求

思路一：模拟迭代

依次判断每个节点是否合法：
- 首先找出结果的根，若原根小了就拉右边的过来，大了拉左边的过来做新根；
- 然后分别判断左右子树的大小，由于二叉搜索树的性质，子树只需要判断一边就好：
  - 左子树判断是否>low，合法就向左下走，不合法往右下；
  - 右子树判断是否<high，合法就向右下走，不合法往左下。

Java

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        while (root != null && (root.val < low || root.val > high)) // 确定原根是否合法
            root = root.val < low ? root.right : root.left;
        TreeNode res = root;
        while (root != null) {
            while (root.left != null && root.left.val < low)
                root.left = root.left.right;
            root = root.left;
        }
        root = res;
        while (root != null) {
            while (root.right != null && root.right.val > high)
                root.right = root.right.left;
            root = root.right;
        }
        return res;
    }
}

时间复杂度：O(n)
空间复杂度：O(1)

C++

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        while (root != nullptr && (root->val < low || root->val > high)) // 确定原根是否合法
            root = root->val < low ? root->right : root->left;
        TreeNode* res = root;
        while (root != nullptr) {
            while (root->left != nullptr && root->left->val < low)
                root->left = root->left->right;
            root = root->left;
        }
        root = res;
        while (root != nullptr) {
            while (root->right != nullptr && root->right->val > high)
                root->right = root->right->left;
            root = root->right;
        }
        return res;
    }
};

时间复杂度：O(n)
空间复杂度：O(1)

思路二：递归

直接用当前函数递归修剪即可：
- 当前值小了放右下（大）的值进去，剪掉当前和左边节点；
- 当前值大了放左下（小）的值进去，剪掉当前和右边节点。
- 然后递归掉下面所有节点。

Java

class Solution {
    public TreeNode trimBST(TreeNode root, int low, int high) {
        if (root == null)
            return null;
        if (root.val < low)
            return trimBST(root.right, low, high);
        else if (root.val > high)
            return trimBST(root.left, low, high);
        root.left = trimBST(root.left, low, high);
        root.right = trimBST(root.right, low, high);
        return root;
    }
}

时间复杂度：O(n)
空间复杂度：O(1)，忽略递归的额外空间开销

C++

class Solution {
public:
    TreeNode* trimBST(TreeNode* root, int low, int high) {
        if (root == nullptr)
            return nullptr;
        if (root->val < low)
            return trimBST(root->right, low, high);
        else if (root->val > high)
            return trimBST(root->left, low, high);
        root->left = trimBST(root->left, low, high);
        root->right = trimBST(root->right, low, high);
        return root;
    }
};

时间复杂度：O(n)
空间复杂度：O(1)，忽略递归的额外空间开销

Rust

今天又见识到了新报错：already borrowed: BorrowMutError，不能把borrow的东西来回随便等，要搞临时中间变量，闭包要关好。

use std::rc::Rc;
use std::cell::RefCell;
impl Solution {
    pub fn trim_bst(root: Option<Rc<RefCell<TreeNode>>>, low: i32, high: i32) -> Option<Rc<RefCell<TreeNode>>> {
        if  root.is_none() {
            return None;
        }
        if root.as_ref().unwrap().borrow().val < low {
            return Solution::trim_bst(root.as_ref().unwrap().borrow().right.clone(), low, high);
        }
        else if root.as_ref().unwrap().borrow().val > high {
            return Solution::trim_bst(root.as_ref().unwrap().borrow().left.clone(), low, high);
        }
        let (l, r) = (Solution::trim_bst(root.as_ref().unwrap().borrow().left.clone(), low, high), Solution::trim_bst(root.as_ref().unwrap().borrow().right.clone(), low, high)); // 要先拎出来
        root.as_ref().unwrap().borrow_mut().left = l;
        root.as_ref().unwrap().borrow_mut().right = r;
        root
    }
}

时间复杂度：O(n)
空间复杂度：O(1)，忽略递归的额外空间开销

以上就是Java C++ 算法题解leetcode669修剪二叉搜索树的详细内容，更多关于Java C++ 算法修剪二叉搜索树的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

在C++中使用HP-Socket
这篇文章主要介绍了C++中简单使用HP-Socket，HP-Socket 是一套通用的高性能 TCP/UDP /HTTP 通信框架，包含服务端组件、客户端组件和 Agent 组件，广泛适用于各种不同应用场景的 TCP/UDP /HTTP 通信系统，下面来看看更具体的介绍吧
2021-11-11
C/C++关于实现CAN信号的获取方法
这篇文章主要介绍了C/C++关于实现CAN信号的获取方法，标准的CAN 数据为8字节，即64位，但是CAN FD的最大数据可为64字节，为512位，其中的帧ID分为标准帧和扩展帧，其中用11位标准帧，用29位表示扩展帧
2023-02-02
浅谈C++中的构造函数分类及调用规则
这篇文章主要介绍了C++中的构造函数分类及调用规则,文中根据参数写出了几种不同类型的构造函数并解释了如何调用,需要的朋友可以参考下
2016-03-03
C语言位域详解及示例代码
本文主要介绍C语言位域的知识，这里整理了相关资料，并附示例代码及详解，有兴趣的小伙伴可以参考下
2016-08-08
C++类中的常数据成员与静态数据成员之间的区别
常数据成员是指在类中定义的不能修改其值的一些数据成员，类似于我们以前学过的常变量，虽然是变量，也有自己的地址，但是一经赋初值，便不能再被修改
2013-10-10
C++ primer类的基础精讲
C++类，是指系统在第一次在程序中遇到一个类时为这个类建立它的所有类变量的拷贝 - 这个类的所有实例共享它的类变量
2022-07-07
C语言实现注册登录系统
这篇文章主要为大家详细介绍了C语言实现注册登录系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2020-12-12
VS2019 更新MSDN并创建快捷方式的实现
这篇文章主要介绍了VS2019 更新MSDN并创建快捷方式的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-03-03
win10环境下C++ vs2015编译opencv249的教程
这篇文章主要介绍了win10环境下C++ vs2015编译opencv249的教程，本文分步骤给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2020-03-03
C++中的并行与并发基础与使用详解
对于多线程来说，这两个概念有很大部分是重叠的。对于很多人来说，它们的意思没有什么区别。其区别主要在于关注点和意图方面(差距甚微)。这两个词都是用来对硬件在同时执行多个任务的方式进行描述的术语，不过并行更加注重性能
2023-02-02