LeetCode669. 修剪二叉搜索树Golang版

最新推荐文章于 2024-11-04 19:37:50 发布

皮卡猴

最新推荐文章于 2024-11-04 19:37:50 发布

阅读量145

点赞数

分类专栏： leetcode刷题文章标签：算法 leetcode 数据结构 go

本文链接：https://blog.csdn.net/flying_monkey_1/article/details/115078308

版权

leetcode刷题专栏收录该内容

252 篇文章 6 订阅

订阅专栏

LeetCode669. 修剪二叉搜索树Golang版

1. 问题描述

给你二叉搜索树的根节点 root ，同时给定最小边界low 和最大边界 high。通过修剪二叉搜索树，使得所有节点的值在[low, high]中。修剪树不应该改变保留在树中的元素的相对结构（即，如果没有被移除，原有的父代子代关系都应当保留）。可以证明，存在唯一的答案。

所以结果应当返回修剪好的二叉搜索树的新的根节点。注意，根节点可能会根据给定的边界发生改变。
在这里插入图片描述

2. 思路

2.1. 递归

确定递归函数参数和返回值

	func trimBST(root *TreeNode, low int, high int) *TreeNode

确定结束条件

	if root == nil {
        return root
    }

确定单层递归逻辑

	if root.Val < low {
        right := trimBST(root.Right, low, high)
        return right
    }
    if root.Val > high {
        left := trimBST(root.Left, low, high)
        return left
    }

    root.Left = trimBST(root.Left, low, high)
    root.Right = trimBST(root.Right, low, high)
    return root

3. 代码

3.1. 递归代码

	/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func trimBST(root *TreeNode, low int, high int) *TreeNode {
    if root == nil {
        return root
    }

    if root.Val < low {
        right := trimBST(root.Right, low, high)
        return right
    }
    if root.Val > high {
        left := trimBST(root.Left, low, high)
        return left
    }

    root.Left = trimBST(root.Left, low, high)
    root.Right = trimBST(root.Right, low, high)
    return root
}