题解 | #重量级的一层#

知识点：

树的层序遍历

解题思路：

用一个queue存储每一层，遍历当前层的queue，一边算出sum一边将下一层节点加入queue。每遍历完一层就判断是否更新res

语言：

Golang

package main
import . "nc_tools"
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * 
 * @param root TreeNode类 
 * @return int整型
*/
func maxLevelSum( root *TreeNode ) int {
    // write code here
    depth:=1
    queue:=[]*TreeNode{root}
    maxSum:=0
    res:=1
    for len(queue)>0{
        size:=len(queue)
        sum:=0
        for i:=0;i<size;i++{
            cur:=queue[0]
            sum+=cur.Val
            queue = queue[1:]
            if cur.Left != nil{
                queue = append(queue, cur.Left)
            }
            if cur.Right !=nil{
                queue = append(queue, cur.Right)
            }
        }
        if sum > maxSum{
            maxSum =sum
            res = depth
        }
        depth++
    }
    return res
}