Sum Root to Leaf Numbers
Total Accepted: 43960 Total Submissions: 144900My Submissions
Question Solution
Given a binary tree containing digits from 0-9
only, each root-to-leaf path could represent a number.
An example is the root-to-leaf path 1->2->3
which represents the number 123
.
Find the total sum of all root-to-leaf numbers.
For example,
1 / \ 2 3
The root-to-leaf path 1->2
represents the number 12
.
The root-to-leaf path 1->3
represents the number 13
.
Return the sum = 12 + 13 = 25
.
分析:计算每条路经表示的数字的和,采用栈数据结构,存储路经过程中每一节点,以便遍历所有结果。遍历时,当使用完一个节点左子树,可以在这个位置处置空,在回溯时,不考虑为空的点,这样不会重复访问节点
public class Solution {
public int sumNumbers(TreeNode root) {
if(root==null)
return 0;
TreeNode x=root;
Stack<TreeNode> stack=new Stack<TreeNode>();
int sum=0;
int path=0;
stack.push(x);
path=path*10+x.val;
while(!stack.isEmpty()){
TreeNode y=stack.peek();
//stack.pop();
if(y.left!=null)
{
TreeNode z=y.left;
y.left=null;
stack.pop();
stack.push(y);
stack.push(z);
path=path*10+z.val;
}
else if(y.right!=null)
{
TreeNode z=y.right;
y.right=null;
stack.pop();
stack.push(y);
stack.push(z);
path=path*10+z.val;
}
else
{
sum=sum+path;
while(!stack.isEmpty())
{
TreeNode z=stack.peek();
if(z.left==null&&z.right==null)
{
path=path/10;
stack.pop();
}
else
break;
}
}
}
return sum;
}
}
转载于:https://blog.51cto.com/7061299/1652655