2641. Cousins in Binary Tree II
给一个二元树的root
把这棵树的所有node的value换成sum of all its cousins' values
cousin就是在同一个depth,但是parent root不同的nodes
思路:
先用bfs记录每一个depth的sum
接着用depth
看这个node有没有左、右子节点
有的话就把子节点那个depth的sum扣掉该node子节点的value然后传下去
这样就可以得到答案了
golang code:
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
func replaceValueInTree(root *TreeNode) *TreeNode {
queue := []*TreeNode{root}
rec := []int{}
for len(queue) > 0 {
cnt := len(queue)
sum := 0
for cnt > 0 {
node := queue[0]
queue = queue[1:]
sum += node.Val
if node.Left != nil {
queue = append(queue, node.Left)
}
if node.Right != nil {
queue = append(queue, node.Right)
}
cnt