LeetCode探索(78):1022. 从根到叶的二进制数之和

120 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第6天,点击查看活动详情

题目

给出一棵二叉树,其上每个结点的值都是 01 。每一条从根到叶的路径都代表一个从最高有效位开始的二进制数。

  • 例如,如果路径为 0 -> 1 -> 1 -> 0 -> 1,那么它表示二进制数 01101,也就是 13

对树上的每一片叶子,我们都要找出从根到该叶子的路径所表示的数字。

返回这些数字之和。题目数据保证答案是一个 32 位 整数。

示例 1:

img

输入: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 仅为 01

思考

本题难度简单。

首先是读懂题意。对于二叉树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)。

参考