637. 二叉树的层平均值（层级遍历）

我就是我 2023-10-05 03:22 46阅读 0赞

#### 637. 二叉树的层平均值（层级遍历） ####

*  题目
 *  解题思路
 *  代码

## 题目 ##

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIxMDQwNTU5_size_16_color_FFFFFF_t_70]

## 解题思路 ##

很容易想到要用层级遍历。关键点是要知道用`queue.size()`来获取该层有多少节点。然后用一个`sum`来保存和 ，进而求出平均值。

## 代码 ##

class Solution {
        
        List<Double> list = new ArrayList<>();
    
        public List<Double> averageOfLevels(TreeNode root) {
        
            
            /*
            因为题目说是非空，所以不用判断
            if(root==null){
                return null;
            }
            */
            Queue<TreeNode> queue = new LinkedList<>();
    
            queue.offer(root);
            while (!queue.isEmpty()) {
        
                double sum = 0.00;
                int count = queue.size();//记录这一层有几个节点
                for (int i = 0; i < count; i++) {
        
                    TreeNode cur = queue.poll();
                    sum += cur.val;//求出当前层的和
    
                    //去寻找下一层
                    if (cur.left != null) {
        
                        queue.offer(cur.left);
                    }
                    if (cur.right != null) {
        
                        queue.offer(cur.right);
                    }
                }
    
                double res = sum / count;
                list.add(res);
    
    
            }
            return list;
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIxMDQwNTU5_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20210116223808980.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzIxMDQwNTU5,size_16,color_FFFFFF,t_70