在Java中返回树形结构可以使用递归方法来实现

飞熊 • 2023年12月19日下午9:24 • 后端漫谈 • 阅读 420

在Java中返回树形结构可以使用递归方法来实现。

以下是一个简单的例子：

1.首先定义一个表示树的结构体(TreeNode)，它包含节点的ID、名称、父节点ID和子节点列表：

Copy codepublic class TreeNode {
    private int id;
    private String name;
    private int parentId;
    private List<TreeNode> children;

    // getter and setter methods
}

2.然后编写一个在树形结构中查找节点的方法，它接受一个根节点ID和节点名称参数：

Copy codepublic TreeNode findNode(int rootId, String name) {
    //查找根节点
    TreeNode rootNode = findRoot(rootId);

    //在根节点的子节点中查找匹配的节点
    TreeNode matchedNode = findNode(rootNode.getChildren(), name);

    return matchedNode;
}

private TreeNode findRoot(int id) {
    // TODO: 根据ID查找根节点
}

private TreeNode findNode(List<TreeNode> nodes, String name) {
    for (TreeNode node : nodes) {
        // 如果节点名称匹配，则返回这个节点
        if (node.getName().equals(name)) {
            return node;
        }
        //如果子节点非空，则递归查找子节点
        if (node.getChildren() != null && !node.getChildren().isEmpty()) {
            TreeNode matchedNode = findNode(node.getChildren(), name);
            if (matchedNode != null) {
                return matchedNode;
            }
        }
    }
    // 如果没找到，则返回null
    return null;
}

在这个方法中，我们先查找根节点，然后在根节点的子节点中递归查找匹配的节点。如果找到匹配的节点，则返回这个节点，否则返回null。

最后，您可以使用以下代码来构建树形结构并返回整棵树：

Copy codepublic TreeNode buildTree() {
    // TODO: 从数据源中获取所有节点,并且初始化根节点
    TreeNode rootNode = new TreeNode();
    rootNode.setId(0);
    rootNode.setParentId(-1);

    // 根据根节点的ID递归构建整棵树
    buildTree(rootNode);

    return rootNode;
}

private void buildTree(TreeNode parentNode) {
    // 根据父节点的ID获取子节点列表
    List<TreeNode> children = getChildrenNodes(parentNode.getId());

    // 将子节点列表设置到父节点中
    parentNode.setChildren(children);

    // 对于每个子节点，递归地构建子树
    for (TreeNode childNode : children) {
        buildTree(childNode);
    }
}