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