LeetCode Average of Levels in Binary Tree问题

Given a non-empty binary tree, return the average value of the nodes on each level in the form of an array.

Example 1:

Input:
    3
   / \
  9  20
    /  \
   15   7
Output: [3, 14.5, 11]
Explanation:
The average value of nodes on level 0 is 3,  on level 1 is 14.5, and on level 2 is 11. Hence return [3, 14.5, 11].

Note:

1. The range of node's value is in the range of 32-bit signed integer.

二叉树的问题，自然想到了二叉树的遍历。这个题是用集合来存储左右节点的平均值。很自然的想到了广度遍历算法。

给出一个广度遍历二叉树的代码：

/** 
     * 广度优先遍历二叉树，又称层次遍历二叉树 
     * @param node 
     */  
    public void breadthFirstTraverse(Node<E> root) {  
        Queue<Node<E>> queue = new LinkedList<Node<E>>();  
        Node<E> currentNode = null;  
        queue.offer(root);  
        while (!queue.isEmpty()) {  
            currentNode = queue.poll();  
            System.out.print(currentNode.value + " ");  
            if (currentNode.left != null)  
                queue.offer(currentNode.left);  
            if (currentNode.right != null)  
                queue.offer(currentNode.right);  
        }  
    }  

就是使用一个队列来说实现的。那么这道题就比较好做了。只要得到每一次广度遍历时节点的个数就可以，在LeetCode上找了一种方法：

List<Double> result = new ArrayList<>();
        Queue<TreeNode> q = new LinkedList<>();

        if(root == null) return result;
        q.add(root);
        while(!q.isEmpty()) {
            int n = q.size();
            double sum = 0.0;
            for(int i = 0; i < n; i++) {
                TreeNode node = q.poll();
                sum += node.val;
                if(node.left != null) q.offer(node.left);
                if(node.right != null) q.offer(node.right);
            }
            result.add(sum / n);
        }
        return result;

我的思路是找一个数组来接收这个所有节点，然后前一个和后一个相加在除以2。这种方法没有AC，当只有一个节点的时候集合中总是有一个为空的多余项。