leetcode637. 二叉树的层平均值

题目：leetcode637. 二叉树的层平均值

描述

给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。

思路

了解层次遍历的都知道，这道题并不困难，只需要一层层遍历相加，最后求平均值。要注意的是，如果你这里使用一个变量来存储一层节点的和，然后再进行求平均值的，那么避免相加后发生溢出，则不可以sum的类型不可以使用int，而是使用long或者double。

代码

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

public class Solution {
public List averageOfLevels(TreeNode root)
{
if(root==null)
return new ArrayList<>();
List result=new ArrayList<>();
Deque deque=new LinkedList<>();
deque.offer(root);
int length;
long sum;
while (!deque.isEmpty())
{
length=deque.size();
sum=0;
for (int i = 0; i < length; i++) {
TreeNode node=deque.poll();
sum+=node.val;
if(node.left!=null)
deque.offer(node.left);
if(node.right!=null)
deque.offer(node.right);
}
result.add(sum/(double)length);
}
return result;
}
}