java 递归查询所有子节点id-CFANZ编程社区

在多层次的数据结构中，如树或图，经常需要查询一个节点下的所有子节点。本文将探讨如何使用Java实现递归查询所有子节点的ID，这对于处理具有层级关系的数据结构特别有用。

数据模型

首先，定义一个简单的节点类Node，该类包含节点ID、父节点ID和子节点列表作为其属性。

public class Node {
    private int id;
    private int parentId;
    private List<Node> children;

    // 构造函数、Getter和Setter省略
}

递归查询实现

递归方法的核心思想是从根节点开始，遍历其子节点，并对每个子节点重复此过程，直到到达叶子节点。下面是这一递归查询逻辑的实现：

import java.util.ArrayList;
import java.util.List;

public class TreeUtils {

    /**
     * 递归查询一个节点下的所有子节点ID
     * @param rootNode 起始节点
     * @return 所有子节点的ID列表
     */
    public static List<Integer> findAllChildNodeIds(Node rootNode) {
        List<Integer> childNodeIds = new ArrayList<>();
        addAllChildNodeIds(rootNode, childNodeIds);
        return childNodeIds;
    }

    private static void addAllChildNodeIds(Node node, List<Integer> childNodeIds) {
        if (node == null) {
            return;
        }
        List<Node> children = node.getChildren();
        if (children != null) {
            for (Node child : children) {
                childNodeIds.add(child.getId());
                addAllChildNodeIds(child, childNodeIds); // 递归调用
            }
        }
    }
}

在这个实现中，findAllChildNodeIds方法是公开调用的入口，它初始化一个空的ID列表，然后调用私有方法addAllChildNodeIds来填充这个列表。递归的停止条件是当前节点为null或其子节点列表为空。

使用示例

以下是如何使用上述工具类的一个简单示例：

public class Main {
    public static void main(String[] args) {
        // 构造节点关系
        Node root = new Node();
        root.setId(1);
        
        Node child1 = new Node();
        child1.setId(2);
        
        Node child2 = new Node();
        child2.setId(3);
        
        root.setChildren(Arrays.asList(child1, child2));
        
        // 查询所有子节点ID
        List<Integer> allChildNodeIds = TreeUtils.findAllChildNodeIds(root);
        
        // 输出结果
        System.out.println("All Child Node IDs: " + allChildNodeIds);
    }
}

以上示例首先创建了一个简单的树形结构，然后使用TreeUtils.findAllChildNodeIds方法查询根节点下的所有子节点ID，并打印这些ID。