Leetcode129. 求根到叶子节点数字之和

最新推荐文章于 2025-12-19 16:34:09 发布

原创最新推荐文章于 2025-12-19 16:34:09 发布 · 182 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

Leetcode 专栏收录该内容

214 篇文章

订阅专栏

本文详细解析了LeetCode129题“求根到叶子节点数字之和”的算法思路，介绍了通过递归和迭代两种方法实现的解决方案，帮助读者理解和掌握二叉树遍历与数字计算的技巧。

Leetcode129. 求根到叶子节点数字之和

题目：
给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。

例如，从根到叶子节点路径 1->2->3 代表数字 123。
计算从根到叶子节点生成的所有数字之和。
说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
    1
   / \
  2   3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此，数字总和 = 12 + 13 = 25.

题解：
递归+迭代
java代码：

public static int sum = 0;

    /**
     * 递归
     *
     * @param root
     * @return
     */
    public static int sumNumbers(TreeNode root) {

        if (root == null) {
            return 0;
        }

        helper(root, 0);
        return sum;
    }

    private static void helper(TreeNode root, int father) {
        if (root == null) return;
        int cur = root.value + father * 10;

        if (root.left == null && root.right == null) {
            sum = sum + cur;
        }
        helper(root.left, cur);
        helper(root.right, cur);

    }


    /**
     * 迭代
     *
     * @param root
     * @return
     */
    public static int sumNumbers2(TreeNode root) {
        int sum = 0;
        if (root == null) return sum;
        Stack<TreeNode> nodeStack = new Stack<>();
        Stack<Integer> numStack = new Stack<>();
        nodeStack.push(root);
        numStack.push(0);

        while (!nodeStack.isEmpty()) {
            TreeNode node = nodeStack.pop();
            Integer num = numStack.pop();
            int cur = num * 10 + node.value;
            if (node.left == null && node.right == null) {
                sum += cur;
            }
            if (node.left != null) {
                nodeStack.push(node.left);
                numStack.push(cur);
            }
            if (node.right != null) {
                nodeStack.push(node.right);
                numStack.push(cur);
            }
        }
        return sum;
    }