简单算法 二叉树根节点到叶子节点的所有路径和(java)
描述
给定一个仅包含数字\ 0-9 0−9 的二叉树,每一条从根节点到叶子节点的路径都可以用一个数字表示。
例如根节点到叶子节点的一条路径是1\to 2\to 31→2→3,那么这条路径就用\ 123 123 来代替。
找出根节点到叶子节点的所有路径表示的数字之和
例如:
这颗二叉树一共有两条路径,
根节点到叶子节点的路径 1\to 21→2 用数字\ 12 12 代替
根节点到叶子节点的路径 1\to 31→3 用数字\ 13 13 代替
所以答案为\ 12+13=25 12+13=25
public int sumNumbers (TreeNode root) {
if(root == null) return 0;
Stack<TreeNode> s1 = new Stack<TreeNode>();
Stack<Integer> s2 = new Stack<Integer>();
int sum = 0;
s1.push(root);
s2.push(root.val);
while(!s1.isEmpty()){
TreeNode node = s1.pop();
int value = s2.pop();
if(node.left == null && node.right == null){
sum = sum + value;
}else{
if(node.right != null) {
s1.push(node.right);
s2.push(value * 10 + node.right.val);
}
if(node.left != null) {
s1.push(node.left);
s2.push(value * 10 + node.left.val);
}
}
}
return sum;
}
递归解法:
public int sumNumbers (TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
test(root, 0 , list);
int sum = 0;
for(int i = 0; i < list.size(); i++){
sum = sum + list.get(i);
}
return sum;
}
public void test(TreeNode root,int value,ArrayList<Integer> list){
if(root == null) return;
value = root.val + 10 * value;
if(root.left == null && root.right == null){
list.add(value);
}
test(root.left, value, list);
test(root.right, value, list);
}