给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
示例 1:
输入:
3
/ \
9 20
/ \
15 7
输出:[3, 14.5, 11]
解释:
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。
提示:
节点值的范围在32位有符号整数范围内。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/average-of-levels-in-binary-tree
着作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解:简单二叉树层次遍历和广度搜索的配合
1、建立队列用于保存每一层的节点;
2、将每一层的节点值相加,并除以总节点数,得到二叉树的当前层的平均值;
3、利用当前层的节点拓展出下一层的所有节点,并将所有节点加入队列,再进行第二个步骤,直到队空。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<Double> averageOfLevels(TreeNode root) {
List<Double> res=new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<>();
queue.offer(root);
while(queue.size()!=0){
int size=queue.size();
int cnt=0;
double sum=0;
while(cnt<size){
TreeNode front=queue.remove();
sum+=front.val;
cnt++;
if(front.left!=null) queue.offer(front.left);
if(front.right!=null) queue.offer(front.right);
}
res.add(sum/size);
}
return res;
}
}