重构一棵树的方案数

/*
题目思路：
1. 利用 Java Stream API 对 pairs 进行处理。
2. 构建图和节点入度的关系。
3. 确定根节点，并检查图的连通性和有效性。
*/
import java.util.*;
import java.util.stream.Collectors;

public class SolutionStream {
    public int checkWays(int[][] pairs) {
        Map<Integer, Set<Integer>> graph = Arrays.stream(pairs)
            .flatMapToInt(Arrays::stream)
            .boxed()
            .distinct()
            .collect(Collectors.toMap(k -> k, v -> new HashSet<>()));

        Map<Integer, Long> indegree = Arrays.stream(pairs)
            .flatMapToInt(Arrays::stream)
            .boxed()
            .collect(Collectors.groupingBy(e -> e, Collectors.counting()));

        Arrays.stream(pairs).forEach(pair -> {
            graph.get(pair[0]).add(pair[1]);
            graph.get(pair[1]).add(pair[0]);
        });

        int root = graph.keySet().stream()
            .filter(node -> indegree.get(node) == graph.size() - 1)
            .findFirst()
            .orElse(-1);

        if (root == -1) return 0;
        return dfs(graph, new HashSet<>(), root) ? 1 : 2;
    }

    private boolean dfs(Map<Integer, Set<Integer>> graph, Set<Integer> visited, int node) {
        visited.add(node);
        return graph.get(node).stream()
            .allMatch(neighbor -> visited.contains(neighbor) || dfs(graph, visited, neighbor)) && visited.size() == graph.size();
    }
}