【LeetCode算法修炼+动画演示】【二叉树的路径和】—— 129. 求根到叶子节点数字之和

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

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

例如，从根到叶子节点路径 1->2->3代表数字 123。

计算从根到叶子节点生成的所有数字之和。

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

示例 1:

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

示例 2:

输入: [4,9,0,5,1]
    4
   / \
  9   0
 / \
5   1
输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此，数字总和 = 495 + 491 + 40 = 1026.

要点

1. 假设你非常暴力，把每一条路径数组都记录后求解也是可以的，就是性能跟不上…这里不过多介绍了。

2. 理解递归的特性，它记录当层递归的变量。
   如果你求出了最后一层的数字是 495，在返回上一层的时候系统会帮你把 49记录下来
   也就是说 4 -> 49 -> 495 返回的时候就是 495 -> 49 -> 4。

解法

代码片段

public int sumNumbers(TreeNode root) {
	if (root == null) {//0.终止条件
		return 0;
	}
	int[] result = new int[1];//1.记录结果
	sumNumbersHelper(root, 0, result, 0);//3.求解结果
	return result[0];//2.返回结果
}

public void sumNumbersHelper(TreeNode root, int preSum, int[] ints, int h) {
	if (root == null) {//4.终止条件
		return;
	}
	int nSum = preSum * 10 + root.val;//5.到达当前层的数字是 上一层和*10+当前数字
	if (root.left == null && root.right == null) {//8.当到达叶子节点的时候将这个数字统计到结果
		ints[0] += nSum;
	}
	sumNumbersHelper(root.left, nSum, ints, h + 1);//6.计算左子树的和
	sumNumbersHelper(root.right, nSum, ints, h + 1);//7.计算右子树的和
}

---

结尾

1.博客地址

2.源代码仓库

如果你在代码里看到了用 数字标记的注释 如 //1.xxx 这是我写代码的顺序，希望能给你一点启发。

---

😁😁😁制作动画过程不易，请大家顺手点赞收藏咯 谢谢~😁😁😁 有其它题目不理解的也可以一起学习，如有错误欢迎指出~