Golang:
思路:二叉树,层序遍历即可
代码如下:
func averageOfLevels(root *TreeNode) []float64 {
var res []float64
if root!=nil{
stack:=[]*TreeNode{root}
for len(stack)!=0{
level:=len(stack)
num:=0
for i:=0;i<level;i++{
num+=stack[i].Val
if stack[i].Left!=nil{
stack=append(stack,stack[i].Left)
}
if stack[i].Right!=nil{
stack=append(stack,stack[i].Right)
}
}
res=append(res,float64(num)/float64(level))
stack=stack[level:]
}
}
return res
}
有疑问加站长微信联系(非本文作者)
![](https://static.golangjob.cn/static/img/footer.png?imageView2/2/w/280)