趣题集.Leetcode 1022

给出一棵二叉树，其上每一个结点的值都是 0 或 1 。每一条从根到叶的路径都表明一个从最高有效位开始的二进制数。例如，若是路径为 0 -> 1 -> 1 -> 0 -> 1，那么它表示二进制数 01101，也就是 13 。

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

以 10^9 + 7 为模，返回这些数字之和。

示例：

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

提示：

树中的结点数介于 1 和 1000 之间。 node.val 为 0 或 1 。

思路分析: 经过DFS+位移运算能够轻松解决问题.当处于根节点时，数字是1，遍历左子节点，将1左移一位再与左子节点的值进行或运算，此时结果为二进制的'10'，也就是十进制的2，当遍历到第3层左子树时，再将2左移一位或上该节点的值，即为'100'，也就是4.碰到叶子节点则返回，而后遍历完全部的路径以后求和即为最后的结果.

代码以下:

public class Solution extends LeetCode {
    int result = 0;
    public int sumRootToLeaf(TreeNode root) {
        reverse(root, 0);
        return result;
    }

    public void reverse(TreeNode node, int num) {
        if (node == null) {
            return;
        }

        num = num << 1 | node.val;
        if (node.left == null && node.right == null) {
            result = result + num;
            return;
        }

        reverse(node.left, num);
        reverse(node.right, num);
    }
}
复制代码