java通用树形结构构建

江南烟雨中_

已于 2024-03-06 15:34:08 修改

阅读量484

点赞数 7

分类专栏： springboot 文章标签： java 开发语言

于 2023-12-11 15:26:17 首次发布

本文链接：https://blog.youkuaiyun.com/python_cdn/article/details/134927314

版权

springboot 专栏收录该内容

2 篇文章

订阅专栏

1.创建节点bean实体

package com.springboot.util;


import lombok.Data;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 * 树节点，所有需要实现树节点的，都需要继承该类
 * @author zhengj.wei
 */
@Data
public class TreeNode<T> implements Serializable {
    private static final long serialVersionUID = 1L;
    /**
     * 主键
     */
    private String id;
    /**
     * 上级ID
     */
    private String parentId;
    /**
     * 子节点列表
     */
    private List<T> children = new ArrayList<>();
}

2.构建树形结构的函数

package com.springboot.util;


import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * 树形结构工具类，如：菜单、机构等
 *
 */
public class TreeUtils {

    /**
     * 根据pid，构建树节点
     */
    public static <T extends TreeNode<T>> List<T> build(List<T> treeNodes, String parentId) {
        List<T> treeList = new ArrayList<>();
        for (T treeNode : treeNodes) {
            if (parentId.equals(treeNode.getParentId())) {
                treeList.add(findChildren(treeNodes, treeNode));
            }
        }
        return treeList;
    }

    /**
     * 查找子节点
     */
    private static <T extends TreeNode<T>> T findChildren(List<T> treeNodes, T rootNode) {
        for (T treeNode : treeNodes) {
            if (rootNode.getId().equals(treeNode.getParentId())) {
                rootNode.getChildren().add(findChildren(treeNodes, treeNode));
            }
        }
        return rootNode;
    }

}