带有列表的树中的BFS

Question

所以我正在构建这个树，它有1.*个节点，每个节点都有一个列表，这个列表本身可以有1.*个节点。

树的前三个层次我可以很好地构建，但如果我不编写所有层次的代码，它就不会更好地工作，因为这些层次都很愚蠢。解决方法当然是使用某种递归方法和一个BFS。

基本上，在我的TreeBuilder类中，我希望能够调用Tree.GetNodeSatDepth（深度）)并获得该深度的所有节点的列表。问题是我不理解如何实现GetNodeSatDepth（深度）方法。

我找到的所有例子都是关于二叉树的。另一种方法是让addChild方法使用一个深度参数，这样我就可以指定在哪个深度插入子元素。

最终，这就是我想要的：我有一棵有根的树。根有一个包含4个子节点的列表。我想要那四个孩子。为每个子节点生成三个节点。所以对于子0有3个子，子1有3个子，子3有3个子，子4有3个子。等等

也许一个可能的解决方案是在每个节点上都有一个level属性，然后搜索该节点，然后返回它的父节点。因为它的父级应该有一个在搜索节点处的所有节点的列表。

Answer 1

试试这个方法：

static void getNodesAtDepth(Node root, int currentLevel, int level, List<Node> nodes) {
    if(root == null) return;
    if(level == 0) {
        nodes.add(root);
        return;
    }

    if(currentLevel + 1 == level) {
        if(root.getNodeList() != null) {
            nodes.addAll(root.getNodeList());
        }
    }

    if(currentLevel < level) {
        if(root.getNodeList() != null) {
            for(Node node : root.getNodeList()) {
                getNodesAtDepth(node, currentLevel + 1,  level , nodes);
            }
        }
    }
}

如何使用：

List<Node> nodeList = new LinkedList<>();
getNodesAtDepth(root, 0, 2, nodeList);

root当然是树的根。nodeList将在所需级别存储所有节点（在我的例子中是2）。第二个参数始终为0，（用于跟踪当前级别）