Java递归查询树所有下级实现方法

介绍

在Java开发过程中,经常会遇到需要查询树结构的所有下级节点的需求。本文将向刚入行的小白介绍如何使用递归的方式来查询树的所有下级节点。我们将按照以下步骤进行讲解:

  1. 建立树结构的数据模型
  2. 实现递归查询方法
  3. 调用递归查询方法并输出结果

数据模型

首先,我们需要建立一个树结构的数据模型。假设我们的树节点包含以下属性:

  • id:节点唯一标识
  • name:节点名称
  • parentId:父节点id

我们可以使用一个类来表示树节点:

public class TreeNode {
    private String id;
    private String name;
    private String parentId;

    // 省略构造方法、getter和setter
}

递归查询方法

接下来我们将实现递归查询方法。该方法将接收一个节点id作为参数,并返回以该节点为根节点的所有子节点。

public List<TreeNode> queryChildren(String nodeId, List<TreeNode> allNodes) {
    List<TreeNode> children = new ArrayList<>();
    for (TreeNode node : allNodes) {
        if (node.getParentId().equals(nodeId)) {
            children.add(node);
            List<TreeNode> grandchildren = queryChildren(node.getId(), allNodes);
            children.addAll(grandchildren);
        }
    }
    return children;
}

代码解释:

  1. 首先创建一个空的children列表,用于存储查询到的子节点。
  2. 遍历所有节点,如果某个节点的父节点id等于给定的节点id,则将该节点添加到children列表中。
  3. 递归调用queryChildren方法,将当前节点作为参数传入,查询当前节点的所有子节点,并将结果添加到children列表中。
  4. 最后返回children列表,即以给定节点为根节点的所有子节点。

调用递归查询方法

在实际使用中,我们需要提供一个方法来调用递归查询方法,并输出结果。以下是一个示例方法:

public void printChildren(String nodeId, List<TreeNode> allNodes) {
    List<TreeNode> children = queryChildren(nodeId, allNodes);
    for (TreeNode child : children) {
        System.out.println(child.getName());
    }
}

代码解释:

  1. 调用queryChildren方法,传入给定的节点id和所有节点列表。
  2. 获取返回的子节点列表。
  3. 遍历子节点列表,输出每个子节点的名称。

示例

我们来看一个完整的示例,假设我们有以下树结构的节点:

      A
     / \
    B   C
   /   / \
  D   E   F
     /
    G

首先,我们需要创建这些节点对象:

TreeNode nodeA = new TreeNode("A", "Node A", null);
TreeNode nodeB = new TreeNode("B", "Node B", "A");
TreeNode nodeC = new TreeNode("C", "Node C", "A");
TreeNode nodeD = new TreeNode("D", "Node D", "B");
TreeNode nodeE = new TreeNode("E", "Node E", "C");
TreeNode nodeF = new TreeNode("F", "Node F", "C");
TreeNode nodeG = new TreeNode("G", "Node G", "E");

List<TreeNode> allNodes = Arrays.asList(nodeA, nodeB, nodeC, nodeD, nodeE, nodeF, nodeG);

接下来,我们调用printChildren方法查询节点C的所有下级节点,并输出结果:

printChildren("C", allNodes);

运行以上代码,将输出以下结果:

Node E
Node G
Node F

总结

通过本文的介绍,我们学习了如何使用递归的方式来查询树的所有下级节点。首先,我们建立了树结构的数据模型,并实现了递归查询方法。然后,通过调用递归查询方法并输出结果,完成了整个查询过程。递归查询树的所有下级节点是一个常见的需求,在实际开发中非常有用。希望本文能够帮助你理解和应用这个方法。