题解 | #牛群平均重量#

知识点：

树的层序遍历

解题思路：

用一个queue存储每一层，遍历当前层的queue，一边算出sum一边将下一层节点加入queue，最后将sum/size放入结果集，直到queue为空。

语言：

Golang

package main
/*
 * type TreeNode struct {
 *   Val int
 *   Left *TreeNode
 *   Right *TreeNode
 * }
 */

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