JAVA父子表转Tree

@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;
}

将一个 List 转换为树形结构,需要先将 List 中的元素按照父子关系进行排序,然后通过递归构建树形结构。下面是一个 Java 实现的示例代码: ```java public class TreeNode { private int id; private String name; private int parentId; private List<TreeNode> children = new ArrayList<>(); // 省略 getter 和 setter 方法 // 将 List 转换为树形结构 public static List<TreeNode> buildTree(List<TreeNode> nodeList) { List<TreeNode> treeList = new ArrayList<>(); Map<Integer, TreeNode> nodeMap = new HashMap<>(); // 将节点放入 Map 中,方便后续查找 for (TreeNode node : nodeList) { nodeMap.put(node.getId(), node); } // 遍历节点,将子节点添加到节点的 children 列表中 for (TreeNode node : nodeList) { if (node.getParentId() == 0) { // 根节点的 parentId 为 0 treeList.add(node); } else { TreeNode parent = nodeMap.get(node.getParentId()); if (parent != null) { parent.getChildren().add(node); } } } return treeList; } } ``` 示例代码中定义了一个 TreeNode 类,表示树节点,包含了节点的 id、name、parentId 和 children 属性。buildTree 方法接收一个 List<TreeNode> 参数,返回树形结构的根节点列表。 在 buildTree 方法中,首先遍历 nodeList,将节点放入一个 Map 中,方便后续查找。然后再次遍历 nodeList,将子节点添加到节点的 children 列表中。如果当前节点的 parentId 为 0,则说明它是根节点,将其添加到 treeList 中返回。最终,treeList 中包含了所有根节点,每个根节点下面可能有子节点。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值