LeetCode 637. 二叉树的层平均值 JAVA

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

示例 1：

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/average-of-levels-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解：简单二叉树层次遍历和广度搜索的配合
1、建立队列用于保存每一层的节点；
2、将每一层的节点值相加，并除以总节点数，得到二叉树的当前层的平均值；
3、利用当前层的节点拓展出下一层的所有节点，并将所有节点加入队列，再进行第二个步骤，直到队空。

/**
 * 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> res=new ArrayList<>();
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(queue.size()!=0){
            int size=queue.size();
            int cnt=0;
            double sum=0;
            while(cnt<size){
                TreeNode front=queue.remove();
                sum+=front.val;
                cnt++;
                if(front.left!=null) queue.offer(front.left);
                if(front.right!=null) queue.offer(front.right);
            }
            res.add(sum/size);
        }
        return res;
    }
}