每日一题 【每日一题】计算二叉树某层节点之和-Python-20211101

Jack · 2021年11月01日 · 最后由 Jack 回复于 2021年11月02日 · 14 次阅读
本帖已被设为精华帖!

给定一个二叉树,并给出深度,求该深度的节点之和

Jack 将本帖设为了精华贴 11月01日 09:23

参考代码:

class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None


class Solution:
    def levelSum(self, root, level):
        p = []
        self.dfs(root, p, 1, level)
        return sum(p)

    def dfs(self, root, p, dep, level):
        if root is None:
            return
        if dep == level:
            p.append(root.val)
            return
        self.dfs(root.left, p, dep + 1, level)
        self.dfs(root.right, p, dep + 1, level)
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册