二叉搜索树染色

/*
思路：
1. 使用 Java Stream API 对中序遍历进行处理，获取所有节点值。
2. 创建一个 Map 记录每个节点的颜色状态，初始化为蓝色。
3. 使用 Stream 处理每个操作，更新相应范围内节点的颜色。
4. 最后统计红色节点的数量。
*/

import java.util.*;
import java.util.stream.Collectors;

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

public class BSTColoringStream {
    public int getRedNodesCount(TreeNode root, int[][] ops) {
        List<Integer> nodes = new ArrayList<>();
        Map<Integer, Boolean> colorMap = new HashMap<>();
        inOrder(root, nodes);
        nodes.forEach(val -> colorMap.put(val, false)); // Initially all nodes are blue

        Arrays.stream(ops).forEach(op ->
            nodes.stream()
                .filter(val -> val >= op[1] && val <= op[2])
                .forEach(val -> colorMap.put(val, op[0] == 1))
        );

        long redCount = colorMap.values().stream().filter(isRed -> isRed).count();
        return (int) redCount;
    }

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