leetcode 637 二叉树层的平均值
2020-02-04 本文已影响0人
Arsenal4ever
这题没啥好说的,会层次遍历,写这个很容易啊!!!
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def averageOfLevels(self, root):
"""
:type root: TreeNode
:rtype: List[float]
"""
self.answer = []
if not root:
return []
level, nextLevel = [root], []
while level:
tmp = []
for node in level:
tmp.append(node)
if node.left:
nextLevel.append(node.left)
if node.right:
nextLevel.append(node.right)
self.answer.append(sum([t.val for t in tmp]) / float(len(tmp)))
level = nextLevel
nextLevel = []
return self.answer