637.average-of-levels-in-binary-tree

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

示例 1：

提示：

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

解法

class Solution {
		public List<Double> averageOfLevels(TreeNode root) {

			List<Double> res = new ArrayList<>();
			Queue<TreeNode> queue = new LinkedList<>();

			queue.add(root);

			while (!queue.isEmpty()) {

				int size = queue.size();
				long sum = 0;
				for (int i = 0; i < size; i++) {

					TreeNode temp = queue.poll();
					sum += temp.val;
					if (temp.left != null) {
						queue.add(temp.left);
					}

					if (temp.right != null) {
						queue.add(temp.right);
					}

				}

				res.add(sum / (double) size);

			}

			return res;

		}
	}