LC-Average of Levels in Binary Tree

本文介绍了一种计算二叉树每层节点平均值的方法。通过深度优先搜索(DFS)递归算法,实现了对二叉树节点的层次遍历,并计算各层节点值的平均数。该算法使用了辅助的二维数组来存储每层节点值总和及数量,最后计算得到各层的平均值。
# 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]
        """
        s = []


        def dfs(node, depth=0):
            if node:
                if len(s) <= depth:
                    s.append([0, 0])
                s[depth][0] += node.val
                s[depth][1] += 1


                dfs(node.left, depth + 1)
                dfs(node.right, depth + 1)

        dfs(root)
        return [a/float(b) for a, b in s]


Sol = Solution()

t1 = TreeNode(1)
t1.left = TreeNode(2)
t1.right = TreeNode(3)


print Sol.averageOfLevels(t1)

0,较为高效
1,学到的东西:
首先,在定义函数时,可以对函数值进行初始化操作,也就是def dfs(node, depth=0):
这样,在调用时,便可以省略一些。当然,也可以不写,然后在调用时传递参数。

其次,算法的思想是定义一个二维数组,利用二维数组的特性,对点的层数进行标记并计算和。这也是层次遍历的基本思想和简单实现。

最后,简单的遍历二维数组的方式:

[a/float(b) for a, b in s]

#a为s的一维元素,b为s的二维元素,依次进行遍历
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值