给出一棵二叉树,其上每个结点的值都是 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

示例 3:

输入:root = [1]
输出:1

示例 4:

输入:root = [1,1]

输出:3

题解

遍历二叉树,收集从根节点到叶子节点的所有节点值,换算成十进制。

因为无法判断层级深度,所以一般等到叶子节点时,再去计算二进制。这里可以使用位运算 << 左移 更加简便

<< 运算符使指定值的二进制数所有位都左移指定次数,其移动规则:丢弃高位,低位补0即按二进制形式把所有的数字向左移动对应的位数,高位移出(舍弃),低位的空位补零。

  1. // 1的二进制表示为: 00000000 00000000 00000000 00000001
  2. // -----------------------------
  3. // 2的二进制表示为: 00000000 00000000 00000000 00000010
  4. console.log(1 << 1) // 2

比如 6 的二进制 110

第一个遍历的数是 1, sum = 1

第二个遍历的数是 1, 现将 sum 左移 1 位在加上当前的数 sum = sum << 1 + 1 等于 3

第三个遍历的数是 0, 同样加 sum 左移 加上 0, 3 << 1 + 0 = 6

  1. /**
  2. * Definition for a binary tree node.
  3. * function TreeNode(val, left, right) {
  4. * this.val = (val===undefined ? 0 : val)
  5. * this.left = (left===undefined ? null : left)
  6. * this.right = (right===undefined ? null : right)
  7. * }
  8. */
  9. /**
  10. * @param {TreeNode} root
  11. * @return {number}
  12. */
  13. var sumRootToLeaf = function(root) {
  14. let res = 0
  15. const run = (node, sum) => {
  16. sum = (sum << 1) + node.val
  17. if (node.left) {
  18. run(node.left, sum)
  19. }
  20. if (node.right) {
  21. run(node.right, sum)
  22. }
  23. if (!node.left && !node.right) {
  24. res += sum
  25. }
  26. }
  27. run(root)
  28. return res
  29. };

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。