受限条件下可到达节点的数目

// Java Stream solution
// 思路：
// 虽然Java Stream更适合数据流处理，但我们依然可以通过组合多个Stream操作来实现DFS。
// 不过，由于DFS涉及到递归或显式的栈操作，Stream的优势并不明显。
// 这里我们主要还是利用Stream对集合进行过滤和遍历。

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

public class Solution {
    public int reachableNodes(int n, int[][] edges, int[] restricted) {
        List<List<Integer>> graph = IntStream.range(0, n)
                .mapToObj(i -> new ArrayList<Integer>())
                .collect(Collectors.toList());
        for (int[] edge : edges) {
            graph.get(edge[0]).add(edge[1]);
            graph.get(edge[1]).add(edge[0]);
        }

        Set<Integer> restrictedSet = Arrays.stream(restricted).boxed().collect(Collectors.toSet());
        boolean[] visited = new boolean[n];

        return dfs(0, graph, visited, restrictedSet);
    }

    private int dfs(int node, List<List<Integer>> graph, boolean[] visited, Set<Integer> restrictedSet) {
        if (restrictedSet.contains(node) || visited[node]) {
            return 0;
        }
        visited[node] = true;
        return 1 + graph.get(node).stream()
                .filter(neighbor -> !visited[neighbor] && !restrictedSet.contains(neighbor))
                .mapToInt(neighbor -> dfs(neighbor, graph, visited, restrictedSet))
                .sum();
    }
}