代码随想录 637.二叉树的层平均值
思路:求层平均值,直接套用层序遍历模板即可。
附代码:
class Solution {public List<Double> averageOfLevels(TreeNode root) {List<Double> list = new ArrayList<Double>();LinkedList<TreeNode> queue = new LinkedList<TreeNode>();if(root == null){return list;}queue.add(root);while(queue.size()>0){int size = queue.size();double sum = 0.0;for(int i = 0;i<size;i++){TreeNode t = queue.remove();sum += t.val;if(t.left != null){queue.add(t.left);}if(t.right != null){queue.add(t.right);}}list.add(sum/size);}return list;}
}