Re: [闲聊] 每日leetcode

楼主: JIWP (JIWP)   2024-06-25 20:33:30
1038. Binary Search Tree to Greater Sum Tree
给一颗二元树
把所有节点的值加上比他还小节点
然后回传
思路:
没什么难的
就从柚子树开始DFS下去
用一个变量记录到目前为止的总和
这样就好
golang code :
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func bstToGst(root *TreeNode) *TreeNode {
var sum int
dfs(root,&sum)
return root
}
func dfs(node *TreeNode,sum *int){
if node==nil{
return
}
dfs(node.Right,sum)
tmp:=(*sum)
(*sum)+=node.Val
node.Val+=tmp
dfs(node.Left,sum)
}

Links booklink

Contact Us: admin [ a t ] ucptt.com