lc1038-从二叉搜索树到更大和树

题目描述

• 给定一棵BST，给每个节点都加上大于自身数值的节点的值

示例

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

输入：root = [0,null,1]
输出：[1,null,1]
解释：
 0        1
  \   =>   \
   1        1

题解

• 思路：dfs
  • 右 -> 根 -> 左，边遍历边累加
  • 跟 cpp 不同，sum 不能设置成全局变量，需要套一个函数保存 sum，且 sum 不能用值传递

func bstToGst(root *TreeNode) *TreeNode {
    sum := 0
    dfs(root, &sum)
    return root
}

func dfs(p *TreeNode, sum *int) {
    if p == nil { return }

    dfs(p.Right, sum)
    t := p.Val
    p.Val += *sum
    *sum += t
    dfs(p.Left, sum)
}