趣题集.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);
    }
}