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++ 算法修剪二叉搜索树的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

Qt中图片旋转缩放操作的实现
本文主要介绍了Qt中图片旋转缩放操作的实现,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
2024-01-01
一篇文章让你彻底明白c++11增加的变参数模板
C++11的新特性--可变模版参数（variadic templates）是C++11新增的最强大的特性之一,它对参数进行了高度泛化，它能表示0到任意个数、任意类型的参数,这篇文章主要给大家详细介绍了关于c++11增加的变参数模板的相关资料,需要的朋友可以参考下
2021-08-08
C++Vector容器常用函数接口详解
最近我学习了C++中的STL库中的vector容器，对于常用容器，我们不仅要会使用其常用的函数接口，我们还有明白这些接口在其底层是如何实现的。所以特意整理出来一篇博客供我们学习
2022-08-08
深入探索C++中stack和queue的底层实现
这篇文章主要介绍了C++中的stack和dequeue的底层实现,本文通过实例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2023-09-09
c语言实现http下载器的方法
这篇文章主要介绍了c语言实现http下载器的相关知识，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
2021-07-07
C和C++的区别详解
这篇文章主要介绍了C和C++之间的区别，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2021-10-10
C语言数据结构之堆排序详解
堆是计算机科学中一类特殊的数据结构的统称，通常是一个可以被看做一棵完全二叉树的数组对象。而堆排序是利用堆这种数据结构所设计的一种排序算法。本文将通过图片详细介绍堆排序，需要的可以参考一下
2022-03-03
C语言中scanf的用法举例
本节介绍输入函数 scanf 的用法，scanf 和 printf 一样，非常重要，而且用得非常多，所以一定要掌握，这篇文章主要介绍了C++中scanf的用法举例,需要的朋友可以参考下
2022-11-11
c++素数筛选法
本文讲的是筛选法的C++实现，筛选法又称筛法，是求不超过自然数N（N>1）的所有质数的一种方法。据说是古希腊的埃拉托斯特尼（Eratosthenes，约公元前274～194年）发明的，又称埃拉托斯特尼筛子。
2017-05-05
c++仿函数和函数适配器的使用详解
这篇文章主要介绍了c++仿函数和函数适配器的使用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-12-12