Java 构建树结构的公共方法

2023-07-27
福建
本文字数：2634 字
阅读完需：约 9 分钟

一、前提

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

-- pId 比如可以传入 3、样例中的 “二、噢噢噢噢”的 id=“e6ee51485389495cb923a122be800012”。然后构建出来的，就是“二、噢噢噢噢”的下级树

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": []                        }                    ]                }            ]        }    ]}

复制代码

发布于: 刚刚阅读数: 3

原文链接:【http://xie.infoq.cn/article/0ba209b7c9dff134bf2921be0】。

高端章鱼哥

关注

还未添加个人签名 2023-06-19 加入

还未添加个人简介

发布

暂无评论

创作场景

Java 构建树结构的公共方法

一、前提

二、代码

三、使用

1、实现 TreeNode 接口

2、使用

3、样例

高端章鱼哥

评论