给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
返回这些数字之和。题目数据保证答案是一个 32 位 整数。
int res =0;
StringBuilder stringBuilder = new StringBuilder();
public int sumRootToLeaf(TreeNode root) {
if (root == null){
return res;
}
dfs(root);
return res;
}
void dfs(TreeNode node){//整体上是一个前序遍历的框架
if (node == null){ //当前节点为空返回
return;
}
stringBuilder.append(node.val);//将当前节点加入
if (node.left == null && node.right == null){//判断当前节点是不是叶子节点,是就进行计算,进行加和
res += Integer.parseInt(stringBuilder.toString(),2) % (Math.pow(10,9)+7);
}
dfs(node.left);//左遍历
dfs(node.right);//右遍历
stringBuilder.deleteCharAt(stringBuilder.length()-1);//删除当前节点
}
int res =0;
public int sumRootToLeaf(TreeNode root) {
preOrder(root,0);
return res;
}
public void preOrder(TreeNode root,int val){//整体是一个DFS广度优先搜索框架
if (root != null){//当前节点不为空,将值左移位加上当前值,赋给tmp,为当前路径的值
int tmp = (val<<1)+ root.val;
if (root.left == null && root.right == null){//当前节点为叶子节点
res += tmp;//把次叶子节点的路径值,加到总值中
}else {
if (root.left != null){//左遍历
preOrder(root.left,tmp);
}
if (root.right != null){//右遍历
preOrder(root.right,tmp);
}
}
}
}