值等于子节点值之和的节点数量

/*
 * 思路：
 * 1. 使用递归遍历树的每个节点。
 * 2. 使用Java Stream API来计算每个节点的子节点和。
 * 3. 如果当前节点值等于其子节点值的和，则计数器加1。
 * 4. 最终返回计数器的值。
 */

import java.util.stream.Stream;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    private int count = 0;

    public int equalToDescendants(TreeNode root) {
        sumDescendants(root);
        return count;
    }

    private int sumDescendants(TreeNode node) {
        if (node == null) return 0;
        int leftSum = node.left == null ? 0 : sumDescendants(node.left);
        int rightSum = node.right == null ? 0 : sumDescendants(node.right);
        int totalSum = Stream.of(leftSum, rightSum).mapToInt(Integer::intValue).sum();
        if (node.val == totalSum) count++;
        return totalSum + node.val;
    }
}