JAVA父子表转Tree

Man20161217

于 2022-03-08 11:21:29 发布

阅读量275

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/Man20161217/article/details/123349077

JAVA 专栏收录该内容

10 篇文章

订阅专栏

这个博客主要介绍了如何使用Java实现树形数据结构的构建。`TreeUtil`类提供了两个核心方法：`buildTree`用于从列表中构建树结构，通过过滤出根节点和子节点；`setChildList`方法递归地为每个父节点设置子节点列表。整个过程利用了Java 8的Stream API，使得代码更加简洁高效。

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

@UtilityClass
public class TreeUtil {
    /**
     * 构建树
     * @param modelTrees
     * @return
     */
    public <T extends TreeVO<T>> List<T> buildTree(List<T> modelTrees){
        if(CollectionUtils.isEmpty(modelTrees)){
            return new ArrayList<>();
        }
        List<T> rootList = modelTrees.stream().filter(e -> e.getPid().equals(0L)).collect(Collectors.toList());
        List<T> childList = modelTrees.stream().filter(e -> e.getPid()>0L).collect(Collectors.toList());
        rootList.forEach(e->setChildList(e,childList));
        return rootList;
    }

    /**
     * 设置树的子节点
     * @param treeVO 父节点
     * @param trees 待处理列表
     */
    public <T extends TreeVO<T>> void setChildList(T treeVO,List<T> trees){
        if(CollectionUtils.isEmpty(trees)){
            return ;
        }
        List<T> collect = trees.stream().filter(a -> a.getPid().equals(treeVO.getId())).collect(Collectors.toList());
        if(CollectionUtils.isEmpty(collect)){
            return ;
        }
        treeVO.setChildList(collect);
        trees.removeAll(collect);
        for (T e : treeVO.getChildList()) {
            setChildList(e, trees);
        }
    }

}

@Data
@ToString
public class TreeVO<T> {
    public Long id;
    public Long pid;
    public List<T> childList;
}