递归建树parentId,list和树状list互相转换

    /**
     * 使用递归方法建树
     * @param treeNodes
     * @return
     */
    public  List<NetTopologyVo> buildByRecursive( List<NetTopologyVo> treeNodes) {
        List<NetTopologyVo> trees = new ArrayList<NetTopologyVo>();
        for (NetTopologyVo treeNode : treeNodes) {
            if ("0".equals(treeNode.getParentId())) {
                trees.add(findChildren(treeNode,treeNodes));
            }
        }
        return trees;
    }

    /**
     * 递归查找子节点
     * @param treeNodes
     * @return
     */
    public static NetTopologyVo findChildren(NetTopologyVo treeNode,List<NetTopologyVo> treeNodes) {
        treeNode.setChildren(new ArrayList<NetTopologyVo>());

        for (NetTopologyVo it : treeNodes) {
            if(treeNode.getId().equals(it.getParentId())) {
                if (treeNode.getChildren() == null) {
                    treeNode.setChildren(new ArrayList<NetTopologyVo>());
                }
                treeNode.getChildren().add(findChildren(it,treeNodes));
            }
        }
        treeNode.setNum(treeNode.getChildren().size());
        return treeNode;
    }

 

 

 

public class Test(){
 
    private Long id;  
    private Long pid;
    private String name;
    private List<Test> children = new ArrayList<>();
 
    //set,get省略
 
}
 

方法list to treeList

/**
     * list转treeList
     * @param list
     * @return
     */
    public static List<Test> list2tree(List<Test> list) {
        List<Test> result = new ArrayList<>();
 
        Map<Long, Test> hash = list.stream().collect(Collectors.toMap(test -> test.getId(), test -> test));
        for (Test test : list) {
            Test p = hash.get(test.getPid());
            if (p == null) {
                result.add(test);
            } else {
                if (p.getChildren() == null) {
                    p.setChildren(new ArrayList<>());
                }
                p.getChildren().add(test);
            }
        }
        return result;
    }
 

树list转普通list,深度优先递归算法

   /**
     * treeList转list
     * @param list
     * @return
     */
    public static List<Test> tree2list(List<Test> list) {
        List<Test> result = new ArrayList<>();
        for (Test test : list) {
            List<Test> c = test.getChildren();
            result.add(test);
            if (!CollectionUtils.isEmpty(c)) {
                result.addAll(tree2list(c));
                test.setChildren(null);//
            }
        }
        return result;
    }
 

 

  • 3
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值