[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. 所用到的数据结构与算法思想

1. List
2. BFS

6. 总结

BFS算法的基础应用，只用在每层遍历时取平均值输出即可，难点在于如何控制每一层的输出从而进行取平均值的操作。