详解Java构建树结构的公共方法

更新时间：2023年04月16日 09:31:59 作者：小小怪下士 XIA

本文主要介绍了详解Java构建树结构的公共方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧

一、前提

pId需要传入用来确认第一级的父节点，而且pId可以为null。
树实体类必须实现：TreeNode接口
MyTreeVo必须有这三个属性：id、pId、children
可以根据不同需求，配置TreeNode和MyTreeVo中固定的属性

二、代码

定义TreeNode接口

public interface TreeNode {
    String getId();
    String getpId();
    List getChildren();
}

需要将pId作为参数传入，在方法中添加一个pId参数，用于确认第一级的父节点。

这个示例代码中，buildTree方法接收两个参数，一个是泛型类型的参数List，另一个是pId，用于确认第一级的父节点。在找到所有的根节点后，我们需要对每个根节点调用buildChildren方法，递归构建子节点。
以便支持传入的pId作为顶级节点

public class TreeBuilder {
    /**
     * @param nodes 需要构建的nodes，不要继承 TreeNode，且需要有 TreeNode 下的三个方法
     * @param pId 父id。为null则是所有pid为null的是顶级节点；不为null，则是从给的pId往下查询并构建树
     **/
    public static <T extends TreeNode> List<T> buildTree(List<T> nodes, String pId) {
        List<T> roots = new ArrayList<>();
        for (T node : nodes) {
            if (node.getId().equals(pId) || (node.getParentId() == null && (pId == null || pId.equals(node.getId())))) {
                roots.add(node);
            }
        }
        for (T root : roots) {
            buildChildren(root, nodes);
        }
        return roots;
    }
 
    private static <T extends TreeNode> void buildChildren(T node, List<T> nodes) {
        for (T child : nodes) {
            if (child.getParentId() != null && child.getParentId().equals(node.getId())) {
                node.getChildren().add(child);
                buildChildren(child, nodes);
            }
        }
    }
}

构建到几级

如果再增加一个参数，构建到树的第几层该如何实现？
根据需求，如果增加一个参数来控制构建到树的第几层，可以在buildChildren方法中添加一个level参数，用于记录当前节点的层数。在递归构建子节点时，如果当前节点的层数小于指定的层数，则继续递归构建子节点，否则停止递归。以下是修改后的Java示例代码：

public class TreeBuilder {
     
     /**
      * @param level 级别 从1开始。1是最顶层
      **/
    public static <T extends TreeNode> List<T> buildTree(List<T> nodes, String pId, int level) {
        List<T> roots = new ArrayList<>();
        for (T node : nodes) {
            if (node.getId().equals(pId) || (node.getParentId() == null && (pId == null || pId.equals(node.getId())))) {
                roots.add(node);
            }
        }
        for (T root : roots) {
            buildChildren(root, nodes, level, 1);
        }
        return roots;
    }
 
    private static <T extends TreeNode> void buildChildren(T node, List<T> nodes, int level, int currentLevel) {
        if (currentLevel >= level) {
            return;
        }
        for (T child : nodes) {
            if (child.getParentId() != null && child.getParentId().equals(node.getId())) {
                if (node.getChildren() == null) {
                    node.setChildren(new ArrayList<>());
                }
                node.getChildren().add(child);
                buildChildren(child, nodes, level, currentLevel + 1);
            }
        }
    }
}

三、使用

1、实现TreeNode接口

public class MyTreeVo implements TreeNode {
 
    /**
     * 主键
     */
    private String id;
 
    /**
     * 父节点ID
     */
    private String pId;
 
    /**
     * 子级
     */
    private List<MyTreeVo> children = Lists.newArrayList();
 
    //其他属性……
 
    public List<MyTreeVo> getChildren() {
        return children;
    }
 
    public String getId() {
        return id;
    }
 
    public String getpId() {
        return pId;
    }
 
    //其他属性的getter、setter……

2、使用

pId可以传入null，也可以传入需要从哪个节点（X）开始构造的 X的id

List<MyTreeVo> tree = TreeUtilQz.buildTree(vos,null);
//tree就是构建好的树结构数据

3、样例

{
    "data": [
        {
            "id": "e6ee51485389495cb923a122be800011",
            "pId": "",
            "name": "一、钢管钢管",
            "children": [
                {
                    "id": "e6ee51485389495cb923a122be800014",
                    "pId": "e6ee51485389495cb923a122be800011",
                    "name": "（二）嘎嘎嘎嘎嘎",
                    "children": []
                },
                {
                    "id": "e6ee51485389495cb923a122be800013",
                    "pId": "e6ee51485389495cb923a122be800011",
                    "name": "（一）顶顶顶顶",
                    "children": []
                }
            ]
        },
        {
            "id": "e6ee51485389495cb923a122be800012",
            "pId": "",
            "name": "二、噢噢噢噢",
            "children": [
                {
                    "id": "e6ee51485389495cb923a122be800015",
                    "pId": "e6ee51485389495cb923a122be800012",
                    "name": "二的下级",
                    "children": [
                        {
                            "id": "e6ee51485389495cb923a122be800016",
                            "pId": "e6ee51485389495cb923a122be800015",
                            "name": "二的下级的下级",
                            "children": []
                        }
                    ]
                }
            ]
        }
    ]
}

到此这篇关于详解Java构建树结构的公共方法的文章就介绍到这了,更多相关Java构建树结构内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

java如何判断一个数是否是素数(质数)
这篇文章主要介绍了java如何判断一个数是否是素数(质数)，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2021-09-09
一行命令同时修改maven项目中多个module的版本号的方法
这篇文章主要介绍了一行命令同时修改maven项目中多个module的版本号的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2019-06-06
Java中toString()、String.valueOf、(String)强转区别
相信大家在日常开发中这三种方法用到的应该很多，本文主要介绍了Java中toString()、String.valueOf、(String)强转区别，感兴趣的可以了解一下
2021-09-09
Java日常练习题,每天进步一点点(8)
下面小编就为大家带来一篇Java基础的几道练习题(分享)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧，希望可以帮到你
2021-07-07
图书管理系统java代码实现
这篇文章主要为大家详细介绍了java代码实现的图书管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-01-01
带你轻松搞定Java面向对象的编程--数组,集合框架
Java是面向对象的高级编程语言，类和对象是 Java程序的构成核心。围绕着Java类和Java对象，有三大基本特性：封装是Java 类的编写规范、继承是类与类之间联系的一种形式、而多态为系统组件或模块之间解耦提供了解决方案
2021-06-06
Java由浅入深全面讲解方法的使用
方法，也称函数，如果想要重复一段或者多段代码块的使用，可以将这些代码封装成一个方法，方法具体表现为某种行为，使用方法可以提高代码的复用性
2022-04-04
详解eclipse项目中.classpath文件的使用
这篇文章主要介绍了详解eclipse项目中.classpath文件的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-10-10
登录EasyConnect后无法通过jdbc访问服务器数据库问题的解决方法
描述一下近期使用EasyConnect遇到的问题,下面这篇文章主要给大家介绍了关于登录EasyConnect后无法通过jdbc访问服务器数据库问题的解决方法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下
2023-02-02
Java 中使用数组存储和操作数据
本文将介绍Java中常用的数组操作方法,通过详细的示例和解释,帮助读者全面理解和掌握这些方法,具有一定的参考价值,感兴趣的可以了解一下
2023-09-09