题目描述
// 129. 求根节点到叶节点数字之和
// 给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。
// 每条从根节点到叶节点的路径都代表一个数字:
// 例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
// 计算从根节点到叶节点生成的 所有数字之和 。
// 叶节点 是指没有子节点的节点。
题解
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
// DFS / 前序遍历
// 基本上跟【Leetcode】112. 路径总和 和 【Leetcode】113. 路径总和 II
// 很像,在前序遍历模板基础上改一下计算规则就行了。
// path记录路径的数字叠加,res记录不同路径之间的求和。
// 前序遍历前,将当前数字堆叠到path中(高位左移,root.val作为低位插入。
// 当遇到叶子结点,直接将path累加到res中。
// 当递归preOrder结束之后,将原来堆叠到path的root.val去掉(低位去掉)
//
// 执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
// 内存消耗:35.2 MB, 在所有 Java 提交中击败了99.98%的用户
class Solution {
int res = 0;
int path = 0;
public int sumNumbers(TreeNode root) {
preOrder(root);
return res;
}
public void preOrder(TreeNode root) {
if (root == null) {
return;
}
path = path * 10 + root.val;
if (root.left == null && root.right == null) {
res += path;
}
preOrder(root.left);
preOrder(root.right);
path = path / 10;
}
}
执行用时:0 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗:35.8 MB, 在所有 Java 提交中击败了65.92%的用户
通过测试用例:
108 / 108
class Solution {
int path = 0;
int res = 0;
public int sumNumbers(TreeNode root) {
if (root == null)
return 0;
pathSum(root, path);
return res;
}
public void pathSum(TreeNode root, int path) {
if (root == null) {
return;
}
if (root.left == null && root.right == null) {
path = path * 10 + root.val;
res += path;
return;
}
pathSum(root.left, path * 10 + root.val);
pathSum(root.right, path * 10 + root.val);
}
}