值为 1 的节点数

// Leetcode 2445: Count Nodes with Value 1 using Java Stream
// 题目思路:
// 给定一个二叉树，求二叉树中节点值为 1 的节点数量。
// 使用递归函数将所有节点收集到一个列表中，然后利用Java Stream进行过滤和统计。

import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

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

public class Solution {
    public int countNodesWithValueOne(TreeNode root) {
        // 创建一个列表来存储所有节点
        List<TreeNode> nodes = new ArrayList<>();
        // 调用递归函数将所有节点添加到列表中
        collectNodes(root, nodes);
        // 使用流过滤值为1的节点并计数
        return (int) nodes.stream().filter(node -> node.val == 1).count();
    }

    private void collectNodes(TreeNode node, List<TreeNode> nodes) {
        if (node == null) return;
        nodes.add(node);
        collectNodes(node.left, nodes);
        collectNodes(node.right, nodes);
    }
}