0
点赞
收藏
分享

微信扫一扫

637. 二叉树的层平均值

AbrahamW 2022-02-05 阅读 43

637. 二叉树的层平均值:

题目链接 :637. 二叉树的层平均值

题目:
给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。


思路:

1、带层值记录的BFS

  • 利用队列存储根节点,找到下一层其子节点加入队列。
  • 根据队列中存储节点数确定每层子节点个数。
  • 将每层节点按照广度优先顺序加入到子集合。
  • 每层中计算节点值的平均值加入到结果集合中
  1. 时间复杂度:O(n+h)=O(n) n为层序遍历遍历节点数量,h为二叉树高度,进行O(h)次计算。
  2. 空间复杂度:O(n) n为集合列表中元素个数,也为n。

AC代码:

  1. BFS
 class Solution {
    public List<Double> averageOfLevels(TreeNode root) {
        List<Double> res = new LinkedList<>();
        if (root == null) return res;

        Queue<TreeNode> q = new LinkedList<>();
        q.offer(root);
        while (!q.isEmpty()) {
            int size = q.size();
            
            double sum = 0;
            for (int i = 0; i < size; i++) {
                TreeNode cur = q.poll();
                if (cur.left != null) {
                    q.offer(cur.left);
                }
                if (cur.right != null) {
                    q.offer(cur.right);
                }
                sum += cur.val;
            }
            
            res.add(1.0*sum/size);
        }

        return res;
    }
}
举报

相关推荐

0 条评论