python求根节点下的节点数_129求根到叶子节点数字之和

class TreeNode:

def __init__(self, x):

self.val = x

self.left = None

self.right = None

a = TreeNode(1)

b = TreeNode(2)

c = TreeNode(3)

a.left = b

a.right = c

# 这道题用递归来做，很容易就做出来的

class Solution:

def sumNumbers(self, root: TreeNode) -> int:

# 定义一个变量，用来存储最后的和

self.sum_num = 0

# 判断二叉树是否为空，如果为空，直接返回0

if not root : return 0

def dfs(root,num):

# 判断当前节点是否为空

if not root:return

# 根据题意进行相加。

num = num * 10 + root.val

# 注意这里，不能让他走到节点为空的那一层，那样的话就多算一遍。

if not root.left and not root.right:

self.sum_num = self.sum_num + num

return

# 然后分别递归左右子树

dfs(root.left,num)

dfs(root.right,num)

dfs(root,0)

# 最后返回计算出来的结果

return self.sum_num

A = Solution()

print(A.sumNumbers(a))

扫码关注我们

微信号：SRE实战

拒绝背锅 运筹帷幄

×

选择打赏方式：

微信

QQ钱包

支付宝

打赏

打赏

打赏

多少都是心意！谢谢大家！！！

×

选择分享方式：

微信扫一扫，分享朋友圈

Or

手机扫一扫，精彩随身带