当前位置:网站首页>【LeetCode】求根节点到叶节点数字之和

【LeetCode】求根节点到叶节点数字之和

2022-08-10 00:39:00 LawsonAbs

1.题目

在这里插入图片描述

2. 思想

递归

3. 代码

# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
    def __init__(self):
        self.res = 0

    def sumNumbers(self, root: Optional[TreeNode]) -> int:
        self.dfs(root,tmp=0)
        return self.res
    
    # 深搜找出所有叶节点
    # tmp 表示根节点到当前节点的值
    def dfs(self,root,tmp):
        if root is None: # 如果是None,说明到头儿了
            return 
        
        if root.left:
            self.dfs(root.left,tmp*10 + root.val)
        if root.right:
            self.dfs(root.right,tmp*10 + root.val)

        # 如果左右节点都是空,说明当前就是根节点
        if root.left ==None and root.right==None:
            self.res += (tmp*10 + root.val)
原网站

版权声明
本文为[LawsonAbs]所创,转载请带上原文链接,感谢
https://lawson-t.blog.csdn.net/article/details/126257771