从根到叶的二进制数之和

/*
 * 思路：
 * 1. 使用Java Stream的方式计算从根到叶子的路径的值。
 * 2. 我们需要对每个路径计算其二进制数，并累加这些值。
 */

import java.util.stream.Stream;
import java.util.stream.IntStream;

public class Solution {
    public int sumRootToLeaf(TreeNode root) {
        return sum(root, 0);
    }

    private int sum(TreeNode node, int currentSum) {
        if (node == null) {
            return 0;
        }
        currentSum = (currentSum << 1) | node.val;
        if (node.left == null && node.right == null) {
            return currentSum;
        }
        return Stream.of(node.left, node.right)
                     .flatMapToInt(n -> IntStream.of(sum(n, currentSum)))
                     .sum();
    }
}

// TreeNode类定义
class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}