1022. 从根到叶的二进制数之和
给出一棵二叉树,其上每个结点的值都是 0 或 1 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。
例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13 。
对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。
返回这些数字之和。题目数据保证答案是一个 32 位 整数。
示例 1:
输入:root = [1,0,1,0,1,0,1]
输出:22
解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22
示例 2:
输入:root = [0]
输出:0
提示:
树中的节点数在 [1, 1000] 范围内
Node.val 仅为 0 或 1
思路:这是一道可以纯dfs解决的题目,用一个List作为结果集,一直遍历二叉树到叶子节点,把二进制字符串加入到结果集中,然后再遍历结果集把二进制数字转成十进制数字,求和即可
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
List<String> res = new ArrayList<>();
public int sumRootToLeaf(TreeNode root) {
Queue<TreeNode> q = new LinkedList<>();
dfs("",root);
int ans = 0;
for(String r : res){
ans += Integer.parseInt(r,2);
}
return ans;
}
public void dfs(String s, TreeNode t){
if(t == null)
return;
s += t.val;
if(t.left == null && t.right == null){
res.add(s);
return;
}
dfs(s,t.left);
dfs(s,t.right);
}
}