乐趣区

关于leetcode:leetcode树之从根到叶的二进制数之和

本文次要记录一下 leetcode 树之从根到叶的二进制数之和

题目

 给出一棵二叉树,其上每个结点的值都是 0 或 1。每一条从根到叶的门路都代表一个从最高无效位开始的二进制数。例如,如果门路为 0 -> 1 -> 1 -> 0 -> 1,那么它示意二进制数 01101,也就是 13。对树上的每一片叶子,咱们都要找出从根到该叶子的门路所示意的数字。以 10^9 + 7 为模,返回这些数字之和。示例:![](https://assets.leetcode-cn.com/aliyun-lc-upload/uploads/2019/04/05/sum-of-root-to-leaf-binary-numbers.png)

输出:[1,0,1,0,1,0,1]
输入:22
解释:(100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22

 

提醒:树中的结点数介于 1 和 1000 之间。node.val 为 0 或 1。起源:力扣(LeetCode)链接:https://leetcode-cn.com/problems/sum-of-root-to-leaf-binary-numbers
著作权归领扣网络所有。商业转载请分割官网受权,非商业转载请注明出处。

题解

/**
 * 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 {public int sumRootToLeaf(TreeNode root) {return sumNode(root, 0);
    }

    public int sumNode(TreeNode node, int sum) {if (node == null) {return 0;}

        sum = 2 * sum + node.val;

        if (node.left == null && node.right == null) {return sum;}

        return sumNode(node.left, sum) + sumNode(node.right, sum);
    }
}

小结

这里采纳递归的办法,当 node 为 null 时返回 0;之后对 sum 累加以后 node.val;若 node.left 及 node.right 为 null 则返回 sum,否则递归计算 sumNode(node.left, sum) 再累加上 sumNode(node.right, sum)。

doc

  • 从根到叶的二进制数之和
退出移动版