求根节点到叶节点数字之和
题目描述:给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。
每条从根节点到叶节点的路径都代表一个数字:
- 例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。
计算从根节点到叶节点生成的 所有数字之和 。叶节点 是指没有子节点的节点。
示例说明请见LeetCode官网。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解法一:递归法
使用递归法解决此问题,递归过程如下:
- 首先,如果当前节点为null,说明是空树,直接返回;
- 如果当前节点不是nll,将当前节点的值添加到 path 中;
- 然后判断当前节点没有左右子节点,说明是叶子节点,将当前的路径值加到result中,然后返回;
- 如果当前节点的左节点不为空时,递归处理左节点;
- 如果当前节点的右节点不为空时,递归处理右节点。
最后,返回result即为结果值。
import com.kaesar.leetcode.TreeNode;
public class LeetCode_129 {
// 最终的累加值
private static int result = 0;
public static int sumNumbers(TreeNode root) {
sumNumbers(root, "");
return result;
}
/**
* 递归法
*
* @param root
* @param path
*/
private static void sumNumbers(TreeNode root, String path) {
// 如果当前节点为null,说明是空树,直接返回
if (root == null) {
return;
}
// 将当前节点的值添加到 path 中
path += root.val;
// 如果当前节点没有左右子节点,说明是叶子节点,将当前的路径值加到result中,然后返回
if (root.left == null && root.right == null) {
result += Integer.valueOf(path);
return;
}
if (root.left != null) {
// 当前节点的左节点不为空时,递归处理左节点
sumNumbers(root.left, path);
}
if (root.right != null) {
// 当前节点的右节点不为空时,递归处理右节点
sumNumbers(root.right, path);
}
}
public static void main(String[] args) {
TreeNode root = new TreeNode(4);
root.left = new TreeNode(9);
root.right = new TreeNode(0);
root.left.left = new TreeNode(5);
root.left.right = new TreeNode(1);
// 测试用例,期望输出: 1026
System.out.println(sumNumbers(root));
}
}
【每日寄语】 人生就像一场赌局,不可能把把都赢,只要筹码在自己手上,就永远都会有希望。
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。