LeetCode 637. 二叉树的层平均值 JAVA

川长思鸟来 2022-12-18 12:00 105阅读 0赞

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。

示例 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;
        }
    }