[LeetCode] 637.二叉树的层平均值(java实现)
1. 题目
637.二叉树的层平均值
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
2. 读题(需要重点注意的东西)
难点在于如何控制每一层的输出。
3. 解法
解法:BFS
/**
* Definition for a binary tree node.
* public 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 {
Queue<TreeNode> queue = new LinkedList<TreeNode>();
List<Integer> list1 = new ArrayList<Integer>();
List<Double> list2 = new ArrayList<Double>();
public List<Double> averageOfLevels(TreeNode root) {
if(root == null) return list2;
queue.offer(root);
while(!queue.isEmpty()){
int count = queue.size();
while(count>0){
TreeNode node = queue.poll();
list1.add(node.val);
if(node.left!=null) queue.offer(node.left);
if(node.right!=null) queue.offer(node.right);
count--;
}
double average = 0;
for(int i = 0 ; i < list1.size();i++){
average = average + list1.get(i);
}
average = average/list1.size();
list1.clear();
list2.add(average);
}
return list2;
}
}
4. 可能有帮助的前置习题
leetcode.102 二叉树的层次遍历
关于leetcode.102 二叉树的层次遍历的解法可见 BFS
5. 所用到的数据结构与算法思想
6. 总结
BFS算法的基础应用,只用在每层遍历时取平均值输出即可,难点在于如何控制每一层的输出从而进行取平均值的操作。