637. 二叉树的层平均值

简单

题目描述

解题思路：使用队列，层次遍历获取平均值

class Solution {

    List<Double> res = new ArrayList<>();

    public List<Double> averageOfLevels(TreeNode root) {
		List<TreeNode> arr = new ArrayList<>();
		arr.add(root);
		avg(arr);
		return res;
    }
	
	
	public void avg(List<TreeNode> list) {
		if(list.size() == 0) return;
		double sum = 0;
		List<TreeNode> arr = new ArrayList<>();
		for(int i = 0;i < list.size();i++) {
			TreeNode node = list.get(i);
			sum += node.val;
			if(node.left != null) arr.add(node.left);
			if(node.right != null) arr.add(node.right);
		}
		res.add(sum / list.size());
		avg(arr);
	}
}