找到 N 叉树的根节点

/*
 * LeetCode 1506: 找到 N 叉树的根节点
 * 题目描述：给定一个包含 N 个节点的 N 叉树，每个节点都有一个唯一的值和若干个子节点。
 * 要求找到该树的根节点。树中每个节点都可能有多个子节点，根节点是唯一一个不被其他节点指向的节点。
 *
 * 解题思路：
 * 1. 通过流的方式遍历所有节点，记录每个节点的子节点集合。
 * 2. 然后再次流式遍历所有节点，找到一个不在子节点集合中的节点，这个节点即为根节点。
 */

import java.util.*;

class Node {
    public int val;
    public List<Node> children;

    public Node() {}

    public Node(int val) {
        this.val = val;
    }

    public Node(int val, List<Node> children) {
        this.val = val;
        this.children = children;
    }
}

public class Solution {
    public Node findRoot(List<Node> tree) {
        Set<Node> seenChildren = tree.stream()
            .flatMap(node -> node.children.stream())
            .collect(Collectors.toSet());
        // 找到根节点
        return tree.stream()
            .filter(node -> !seenChildren.contains(node))
            .findFirst()
            .orElse(null); // 如果没有找到，返回null
    }
}