LeetCode637. 二叉树的层平均值

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组.

示例 1:

输入:
    3
   / \
  9  20
    /  \
   15   7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3,  第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].

注意：

1. 节点值的范围在32位有符号整数范围内。

---

思路：广度优先遍历二叉树。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
         List<Double> list= new ArrayList<Double>();
	        if(root==null) {
	        	return list;
	        }
	        ArrayDeque<TreeNode> queue=new ArrayDeque<TreeNode>();
	        queue.addLast(root);
	        while(queue.isEmpty()==false) {
	        	int size=queue.size();
	        	long sum=0;
	        	for(int i=0;i<size;i++) {
	        		TreeNode node=queue.removeFirst();
	        		sum+=node.val;
	        		if(node.left!=null) {
	        			queue.addLast(node.left);
	        		}
	        		if(node.right!=null) {
	        			queue.addLast(node.right);
	        		}
	        	}
	        	double ave=(double)sum/size;
	        	list.add(ave);
	        }
	        return list;
    }
}