leetcode和算法----日更

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
上一篇下一篇

猜你喜欢

热点阅读