学习算法,锻炼自我!记录自己的成长过程!
给你一个二叉树的根节点 root ,树中每个节点都存放有一个 0 到 9 之间的数字。每条从根节点到叶节点的路径都代表一个数字:例如,从根节点到叶节点的路径 1 -> 2 -> 3 表示数字 123 。计算从根节点到叶节点生成的 所有数字之和 。叶节点 是指没有子节点的节点。
- 4->9->5 495
- 4->9->1 491
- 4->0 40
结果:495+491+40=1026
这道题需要用到广度优先遍历,因为它其实相当于是一层一层拼接,最后把每个路径的值计算出来,同时还是使用迭代的方式来实现广度优先遍历:
var sumNumbers = function (root) {
if (!root) {
return 0;
};
//这里使用队列来实现广度优先遍历,存储当前节点和当前拼接的数字
const queue = [[root, root.val + ""]];
let res = 0;
while (queue.length) {
let len = queue.length;
while (len--) {
const [r, val] = queue.shift();
if (r.left) {
queue.push([r.left, val + r.left.val]);
};
if (r.right) {
queue.push([r.right, val + r.right.val]);
};
if (!r.left && !r.right) {
res += Number(val);
};
};
};
return res;
};