LC637.二叉树的层平均值

题目

查看题目

解题思路

主要思路：

• 层序遍历
• Double和int之间的转换关系
• 由于使用的是while循环，会对levelsize进行操作，所以最后求平均数的时候，不要再使用levelsize，否则会出现“除0”的错误。

代码

/**
 * 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 {
    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> list = new ArrayList<>();
        Deque<TreeNode> que = new LinkedList<>();
        if (root == null) {
            return list;
        }
        que.offerLast(root);
        
        while (!que.isEmpty()) {
            Double sum = 0.0;
            int levelsize = que.size();
            int caculator_sum = levelsize;

            while (levelsize > 0) {
                // 下面关于新节点的定义要在循环内部进行定义
                // 如果定义在循环外部的话，添加下一层节点的时候只会添加队列首部元素的左右子树
                // 而不会增加后面节点的子树。
                TreeNode node = que.peekFirst();
                Double temp = Double.valueOf(que.pollFirst().val);
                sum += temp;
                // sum += temp;
                if (node.left != null) {
                    que.offerLast(node.left);
                }
                if (node.right != null) {
                    que.offerLast(node.right);
                }
                levelsize--;
            }
            list.add(sum / caculator_sum);// 这里注意不要用和循环里一样的levelsize，否则会出现除0的问题
        }
        return list;
    }
}