leetcode637. 二叉树的层平均值

最新推荐文章于 2024-09-29 23:23:41 发布

伊比利亚圣僧

最新推荐文章于 2024-09-29 23:23:41 发布

阅读量107

点赞数

分类专栏： leetcode 树文章标签： leetcode python

本文链接：https://blog.csdn.net/weixin_42237989/article/details/105508969

版权

leetcode 同时被 2 个专栏收录

15 篇文章 0 订阅

订阅专栏

树

10 篇文章 0 订阅

订阅专栏

感觉BFS还是舒服啊

class Solution(object):
    def averageOfLevels(self, root):
        """
        :type root: TreeNode
        :rtype: List[float]
        """
        self.ans=[]
        self.list=[]
        self.list.append(root)
        while self.list:
            sum=0.0
            l=len(self.list)
            for _ in range(l):
                curr=self.list.pop(0)
                if curr.left:
                    self.list.append(curr.left)
                if curr.right:
                    self.list.append(curr.right)      
                sum+=curr.val
            self.ans.append(sum/l)
        return self.ans