关于累加树:二叉搜索树具有如下的特征,左子树的键值小于根节点,右子树的键值大于根节点。累加树即针对当前节点,计算当前节点键值+比该节点的键值大的节点键值之和。
例如:树中的节点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
}
最后的通过率,时空复杂度还不错。