LeetCode 538 把二叉搜索树转换为累加树

关于累加树:二叉搜索树具有如下的特征,左子树的键值小于根节点,右子树的键值大于根节点。累加树即针对当前节点,计算当前节点键值+比该节点的键值大的节点键值之和。

例如:树中的节点6,根据二叉搜索树的特点不难发现,比它大的节点分别是7和8,将这三个节点的值累加,就得到了节点6对应的累加树的值 6+7+8=21。

1、回溯:看到二叉树的时候,首先想到的就是回溯,观察后比较容易发现通过后续遍历将前面的各个键值与本身键值进行相加即可。在构建后续遍历的过程中遇到了不小的问题,主要在于如何将值进行传递,最后通过 (1)sum 参数:传递搜索树中所有大于当前节点的累加和。(2)返回值:传递以当前为根的子树的累加和。

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func convertBST(root *TreeNode) *TreeNode {
    var subsequent func(r *TreeNode, sum int) int
    subsequent = func(r *TreeNode, sum int) int {
        if r == nil{ 
            return sum
        } else {
            sum1 := subsequent(r.Right, sum)
            r.Val = sum1+r.Val
            sum2 := subsequent(r.Left, r.Val)
            return sum2
        }
    }
    subsequent(root, 0)
    return root 
}

最后的通过率,时空复杂度还不错。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值