Question

我试图找到二叉树中每个级别的平均值。我在做BFS。我试图使用null节点。每当我找到一个虚拟节点时，这意味着我处于该级别的最后一个节点。我面临的问题是我无法使用此方法在树中添加最后一级的平均值。有人可以帮帮我吗？

考虑例子[3,9,20,15,7] 我得到的输出为[3.00000,14.50000]。没有达到最后一级15和7的平均值这是我的代码

/**
* Definition for a binary tree node.
* public class TreeNode {
*     int val;
*     TreeNode left;
*     TreeNode right;
*     TreeNode(int x) { val = x; }
* }
*/

public class Solution {
 public List<Double> averageOfLevels(TreeNode root) {
    List<Double> list = new ArrayList<Double>();
    double sum = 0.0;
    Queue<TreeNode> q = new LinkedList<TreeNode>();
    TreeNode temp = new TreeNode(0);
    q.offer(root);
    q.offer(temp);
    int count = 0;
    while(!q.isEmpty()){
        root = q.poll();
        sum += root.val;

        if(root != temp)
        {
            count++;
            if(root.left != null){
                q.offer(root.left);
            }
            if(root.right != null){
                q.offer(root.right);
            }
        }
        else
        {
            if(!q.isEmpty()){
            list.add(sum / count);
            sum = 0;
            count = 0;
            q.add(temp);           
            }
      }

    }
    return list;
  }
}

Answer 1

我会使用树的递归深度扫描。在每个节点上，我会将值推送到具有一对的映射中。

我没有测试该代码，但它应该是一致的。

void scan(int level, TreeNode n, Map<Integer, List<Integer> m) {
  List l = m.get(level); if (l==null) {
    l = new ArrayList();
    m.put(level, l);
  }
  l.add(n.val);
  int nextLevel = level + 1;
  if (n.left != null) scan(nextLevel, n.left, m);
  if (n.right != null) scan(nextLevel, n.right, m);
}

扫描完成后，我可以计算每个级别的平均值。

for (int lvl in m.keyset()) {
  List l = m.get(lvl);
  // MathUtils.avg() - it is obvious what it should be
  double avg = MathUtils.avg(l);
  // you code here
}

Answer 2

看看这段代码，只要你找到当前级别结束的标记就会执行：

    if(!q.isEmpty()){
        list.add(sum / count);
        sum = 0;
        count = 0;
        q.add(temp);           
    }

此if语句似乎旨在检查您是否已完成树中的最后一行，您可以通过注意队列中不再有与下一级。在这种情况下，您需要更正您不想将虚拟节点添加回队列（这会导致无限循环），但请注意您还没有计算平均值你刚刚完成的那一行。

要解决此问题，您需要计算最后一行的平均值，而不是重新设置队列，如下所示：

   if(!q.isEmpty()){
        q.add(temp);           
   }

   list.add(sum / count);
   sum = 0;
   count = 0;

需要注意一个新的边缘情况，如果树完全是空的，会发生什么。我会让你弄清楚如何从这里开始。祝你好运！

二叉树中每个级别的平均值

2 个答案: