算法学习：637.二叉树的层平均值

二叉树的层平均值

题目链接：力扣题目链接
难度：简单
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
示例
输入：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 TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(){}
     TreeNode(int val){
         this.val = val;
     }
     TreeNode(int val,TreeNode left,TreeNode right){
         this.val = val;
         this.left = left;
         this.right = right;
     }
}

思路

本题就是层序遍历的时候把一层求个总和在取一个均值。

迭代代码

class Solution{
	public List<Double> averageOfLevels(TreeNode root){
		List<Double> result = new ArrayList<>();
		if(root == null) return result;
		Queue<TreeNode> queue = new LinkedList<>();
		queue.offer(root);
		while(!queue.isEmpty()){
			double sum  = 0;
			int len = queue.size();
			for	(int i=0;i<len;i++){
				TreeNode tempNode = queue.poll();
				sum += tempNode.val;
				if(tempNode.left != null)	queue.offer(tempNode.left);
				if(tempNode.right != null)	queue.offer(tempNode.right);
			}
			result.add(sum / len);
		}
		return result;
	}
}