持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第6天,点击查看活动详情
题目
给出一棵二叉树,其上每个结点的值都是 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
思考
本题难度简单。
首先是读懂题意。对于二叉树root = [1,0,1,0,1,0,1],我们从根节点向叶子节点遍历,可以得到路径100、101、110、111,然后,将这些路径视为二进制数,进行求和。
我们使用深度优先搜索对二叉树进行遍历。定义变量res,首先加上根节点的数值,然后遍历左右节点。此时,res先左移一位再加上左右节点的值(也就是视为二进制的求值过程)。当访问到叶子节点时,返回节点的数值、停止遍历。这也就是递归的过程。
该方法总共访问 n 个节点,时间复杂度为O(n),其中 n 是节点数目。
解答
方法一:深度优先搜索
/**
* @param {TreeNode} root
* @return {number}
*/
var sumRootToLeaf = function(root) {
const dfs = (root, val) => {
if (!root) {
return 0
}
val = (val << 1) | root.val // root.val取值0和1,因此异或|或者加号+都可以
if (!root.left && !root.right) {
return val
}
return dfs(root.left, val) + dfs(root.right, val)
}
return dfs(root, 0)
}
// 执行用时:64 ms, 在所有 JavaScript 提交中击败了81.25%的用户
// 内存消耗:43.8 MB, 在所有 JavaScript 提交中击败了20.09%的用户
// 通过测试用例:63 / 63
复杂度分析:
- 时间复杂度:O(n),其中 n 是节点数目。总共访问 n 个节点。
- 空间复杂度:O(n)。