题目地址:
https://leetcode-cn.com/probl...
题目描述:java
给定一个二叉树,它的每一个结点都存放一个 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.
解答:
方法一,自底向上求出全部的路径,而后把路径求和。
java ac代码:node
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public int sumNumbers(TreeNode root) { if(root == null)return 0; List<String> list = getList(root); int ans = 0; for(String s : list) ans += Integer.parseInt(s); return ans; } List<String>getList(TreeNode root) { List<String> ans = new ArrayList(); if(root == null) return ans; List<String>t1 = getList(root.left); List<String>t2 = getList(root.right); for(String s:t1) ans.add(root.val+s); for(String s:t2) ans.add(root.val+s); if(t1.size() == 0 && t2.size() == 0) ans.add(root.val+""); return ans; } }
方法二,自顶向下搜索(深度优先搜索),搜索过程当中维护一个当前累加值,一旦搜索到叶子节点,就将如今的当前累加值加入结果中。这样能够每一个节点只访问一遍,而且没有额外空间消耗。
java ac代码:code
/** * Definition for a binary tree node. * public class TreeNode { * int val; * TreeNode left; * TreeNode right; * TreeNode(int x) { val = x; } * } */ class Solution { public int sumNumbers(TreeNode root) { dfs(root,0); return ans; } int ans = 0; void dfs(TreeNode root,int temp) { if(root == null) return; temp = temp*10+root.val; if(root.left == null && root.right == null) { ans += temp; return; } dfs(root.left,temp); dfs(root.right,temp); } }