28.树形结构工具类

三尺微命一介书生

已于 2022-05-11 16:37:18 修改

阅读量316

点赞数 1

分类专栏： Java工具类文章标签： java

于 2022-05-09 10:11:47 首次发布

zlx

本文链接：https://blog.youkuaiyun.com/weixin_43741092/article/details/124660413

版权

Java工具类专栏收录该内容

30 篇文章

订阅专栏

这篇博客介绍了如何使用Java的Stream API和递归方法，通过泛型参数将列表转换为树形结构。重点讲解了TreeNodeDTO类的设计和listToTree方法的实现，展示了如何处理根节点和子节点关系。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public class TreeUtils {

    /**
     * @param list         全部节点列表
     * @param rootParentId 根节点ID
     * @param <T>          节点数据model
     * @return 树结构数据
     */
    public static <T extends TreeNodeDTO> List<T> listToTree(List<T> list, Long rootParentId) {
        return list.stream()
                .filter(treeNodeDTO -> treeNodeDTO.getParentId().equals(rootParentId))
                .peek(treeNodeDTO -> treeNodeDTO.setChildrenList(listToTree(list, treeNodeDTO.getId())))
                .collect(Collectors.toList());
    }

    public static class TreeNodeDTO {

        /**
         * ID
         */
        private Long id;

        /**
         * 父级ID
         */
        private Long parentId;

        /**
         * 子节点列表
         */
        private List<? extends TreeNodeDTO> childrenList;

        public Long getId() {
            return id;
        }

        public void setId(Long id) {
            this.id = id;
        }

        public Long getParentId() {
            return parentId;
        }

        public void setParentId(Long parentId) {
            this.parentId = parentId;
        }

        public List<? extends TreeNodeDTO> getChildrenList() {
            return childrenList;
        }

        public void setChildrenList(List<? extends TreeNodeDTO> childrenList) {
            this.childrenList = childrenList;
        }
    }
}