leetcode 129 Sum Root to Leaf Numbers

树节点路径求和算法

最新推荐文章于 2025-09-08 16:25:29 发布

MVincent

最新推荐文章于 2025-09-08 16:25:29 发布

阅读量231

点赞数

CC 4.0 BY-SA版权

文章标签： leetcode java python

本文链接：https://blog.youkuaiyun.com/MRxjh/article/details/79620705

本文介绍了一种算法，用于计算二叉树中从根节点到叶子节点的所有路径数值之和。提供了Java和Python两种实现方式，通过递归和迭代的方法遍历树结构并计算总和。

Java:

class Solution {
    public int sumNumbers(TreeNode root) {
        return sumNum(root, 0);
    }
    
    public int sumNum(TreeNode root, int sum){
        if(root == null) return 0;
        if(root.left == null && root.right == null) return sum*10 + root.val;
        return sumNum(root.left, sum*10+root.val) + sumNum(root.right, sum*10+root.val);
    }
}

Python:

class Solution:
    def sumNumbers(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        stack, res = [(root, root.val)], 0
        while stack:
            node, val = stack.pop()
            if not node.left and not node.right:
                res+=val
            if node.left:
                stack.append((node.left, val*10+node.left.val))
            if node.right:
                stack.append((node.right, val*10+node.right.val))
        return res