leetcode-637. 二叉树的层平均值(python)

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

示例 1:

输入:
3
/
9 20
/
15 7
输出: [3, 14.5, 11]
解释:
第0层的平均值是 3, 第1层是 14.5, 第2层是 11. 因此返回 [3, 14.5, 11].
注意：
注意：节点值的范围在32位有符号整数范围内。

    class Solution:
        def averageOfLevels(self, root):
            """
            :type root: TreeNode
            :rtype: List[float]
            """
            def foo(node, data, level):
                if node is None:
                    return
                if len(data) == level:
                    data.append([node.val])
                else:
                    data[level].append(node.val)
                foo(node.left, data, level+1)
                foo(node.right, data, level+1)
            data = []
            foo(root, data, 0)
            out = []
            for datum in data: # each level
                out.append(sum(datum)/len(datum))
            return out