LeetCode 1022 Sum of Root To Leaf Binary Numbers

给予一颗二叉树,每个节点的值只有 0 和 1, 每个根到叶子节点的路径都是一个有效的二进制树, 如: 0 -> 1 -> 1 -> 0 -> 1 , 那么用二进制表示就是 01101 , 对应十进制为 13 .

对于树中的所有叶子节点, 请返回根节点到这些叶子节点的路径所代表的二进制值的和.

例 :

给予树:

     1
   /   \
  0     1
 / \   / \
0   1 0   1

返回: 22. 
说明:  (100) + (101) + (110) + (111) = 4 + 5 + 6 + 7 = 22

解法

采用深度优先遍历, 从跟节点开始, 到下一层的时候, 将父节点的值向左移动一位, 再加上当前节点的值, 直到根节点为止.

如上图中, 1 -> 0 -> 0 这个路径, 从根节点 1 开始, 到第二层, 对 1 左移一位 i = 1 << 1 , 等于 10 , 再加上当前节点的值 0 等于 10 , 再到下一层, 向左移动一位 i = 10 << 1 , 等于 100 .

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    
    int result = 0;
    
    public int sumRootToLeaf(TreeNode root) {
        dfs(root, 0);
        return result;
    }
    
    public void dfs(TreeNode node, int currentVal) {
        if (node != null) {
            currentVal <<= 1;
            currentVal += node.val;
            if (node.left == null && node.right == null) {
                result += currentVal;
            }
            dfs(node.left, currentVal);
            dfs(node.right, currentVal);
        }
    }
}
Runtime: 0 ms, faster than 100.00% of Java online submissions for Sum of Root To Leaf Binary Numbers. Memory Usage: 35.4 MB, less than 100.00% of Java online submissions for Sum of Root To Leaf Binary Numbers.
我来评几句
登录后评论

已发表评论数()

相关站点

+订阅
热门文章