題目

給定一個非空二叉樹的根節(jié)點 root , 以數(shù)組的形式返回每一層節(jié)點的平均值。與實際答案相差 10-5 以內(nèi)的答案可以被接受。

示例 1：

輸入：root = [3,9,20,null,null,15,7]
輸出：[3.00000,14.50000,11.00000]
解釋：第 0 層的平均值為 3,第 1 層的平均值為 14.5,第 2 層的平均值為 11 。
因此返回 [3, 14.5, 11] 。

解

class Solution {public List<Double> averageOfLevels(TreeNode root) {List<Double> result = new ArrayList<>();LinkedList<TreeNode> queue1 = new LinkedList<>();LinkedList<TreeNode> queue2 = new LinkedList<>();queue1.offer(root);Double sum = 0D;int count = 0;while(!queue1.isEmpty()){TreeNode poll = queue1.poll();sum += poll.val;count++;if(poll.left != null){queue2.offer(poll.left);}if(poll.right != null){queue2.offer(poll.right);}if(queue1.isEmpty()){result.add(sum/count);queue1 = queue2;sum = 0D;count = 0;queue2 = new LinkedList<>();}}return result;}
}