求根到叶子节点数字之和（DFS简单实现）

最新推荐文章于 2022-09-08 08:58:41 发布

一群枫子

最新推荐文章于 2022-09-08 08:58:41 发布

阅读量185

点赞数

文章标签：二叉树 leetcode dfs

本文链接：https://blog.youkuaiyun.com/weixin_43857345/article/details/109223209

版权

这篇博客介绍了一个计算二叉树从根节点到叶子节点所有路径数字之和的问题。给定每个节点包含0-9的数字，路径代表一个数字。通过深度优先搜索（DFS）遍历树，累加路径上的数字，最终返回所有叶子节点路径的数字总和。例如，对于输入的二叉树[1,2,3]，输出为25，因为路径1->2和1->3分别代表数字12和13，总和为25。注意，题目强调不需要使用回溯，因为结果存储在基础数据类型int中，不会被其他函数修改。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

给定一个二叉树，它的每个结点都存放一个 0-9 的数字，每条从根到叶子节点的路径都代表一个数字。
例如，从根到叶子节点路径 1->2->3 代表数字 123。计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

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

注意点

注意：该题不需要使用回溯是因为参数sumTemp为基础数据类int，所以sumTemp不会因为其他函数的修改而修改。

实现

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
	// 结果
    int sum = 0;
    public int sumNumbers(TreeNode root) {
        dfs(root, 0);
        return sum;
    }

    public void dfs(TreeNode root, int sumTemp) {
        if (root == null) {
            return;
        }
		
		// 更新路径的值
        sumTemp = sumTemp * 10 + root.val;
        // 判断是否为叶子节点
        if (null == root.left && null == root.right) {
        	//更新结果
            sum += sumTemp;
            return;
        }

		// 递归计算左子树
        dfs(root.left, sumTemp);
        // 递归计算右子树
        dfs(root.right, sumTemp);
    }
}