java实现树形结构还在用递归?看完这个教你如何高效率遍历封装树形结构!

java实现树形结构还在用递归?看完这个教你如何高效率遍历封装树形结构

递归遍历树形结构

以下是递归代码

可以看到我们该代码中的递归方式,需要将集合中的代码不断递归以及封装子节点,相当于每次构建一个节点数据就需要把之前的node集合在遍历一遍。由此看出递归封装树形结构时间复杂度和空间复杂度是 o(n)

 /**
     * 递归封装树形结构
     * @param node
     * @param id
     * @return
     */
    public static List<Tree> buildTree(List<Tree> node, Integer id) {
        List<Tree> res = new ArrayList<>();
        node.forEach(n->{
            if (n.getParentId().equals(id)){
                n.setNodes(buildTree(node,n.getId()));
                res.add(n);
            }
        });
        return res;
    }

hash表实现树形结构

以下是用的是hashMap来实现

通过hash表实现树形结构,只需要两遍循环即可完成封装树形结构
第一遍循环:将所有对象id为key,value为当前对象存入hashMap表中
第二遍循环:使用之前已经存入的hashMap表,通过当前对象的parentId(父级id)获取父级对象,存入当前父级对象中的子级对象里
由此可以看出,使用hash表来实现,只需要两遍循环就可以完成构建出树形结构,效率性能都比较好

public static List<Tree> buildTree(List<Tree> node) {
        List<Tree> res = new ArrayList<>();
        Map<Integer,Tree> map=new HashMap<>();
        node.forEach(n->map.put(n.getId(),n));
        node.forEach(n->{
            //表示不是根目录
            if (n.getParentId()!=0){
                //get当前父级节点
                Tree tree = map.get(n.getParentId());
                //将当前对象添加到当前父级节点的子级节点属性中
                tree.getNodes().add(n);
            }else {
                res.add(n);
            }
        });
        return res;
    }

递归和hash表构建树形结构性能测试

递归实现

设置50000个数据,先看下递归需要多久

在这里插入图片描述

可以看到足足花了五万多毫秒,差不多五十多秒
在这里插入图片描述

hash表实现

在这里插入图片描述
看!使用hash表,六百多毫秒,五万数据耗时一秒不到!
在这里插入图片描述

/** * 根据等级查询类目树 * * @param level * @return */ @Override public List queryCategoryTree(Integer level) { //查询当前级别下类目 List list = categoryDAO.list(level); //组装好的类目树,返回前端 List categoryTree = new ArrayList(); //所有类目 List allDTOList = new ArrayList(); if (CollectionUtils.isEmpty(list)) { return categoryTree; } for (CategoryDO categoryDO : list) { allDTOList.add(new CategoryTreeDTO().convertDOToDTO(categoryDO)); } //当前等级类目 categoryTree = allDTOList.stream().filter(dto -> level.equals(dto.getLevel())).collect(Collectors.toList()); for (CategoryTreeDTO categoryTreeDTO : categoryTree) { //组装类目为树结构 assembleTree(categoryTreeDTO, allDTOList,Constants.CATEGORY_MAX_LEVEL - level); } return categoryTree; } /** * 组装树 * * @param categoryTreeDTO * @param allList * @param remainRecursionCount 剩余递归次数 * @return */ public CategoryTreeDTO assembleTree(CategoryTreeDTO categoryTreeDTO, List allList, int remainRecursionCount) { remainRecursionCount--; //最大递归次数不超过Constants.CATEGORY_MAX_LEVEL-level次,防止坏数据死循环 if(remainRecursionCount < 0){ return categoryTreeDTO; } String categoryCode = categoryTreeDTO.getCategoryCode(); Integer level = categoryTreeDTO.getLevel(); //到达最后等级树返回 if (Constants.CATEGORY_MAX_LEVEL == level) { return categoryTreeDTO; } //子类目 List child = allList.stream().filter(a -> categoryCode.equals(a.getParentCode())).collect(Collectors.toList()); if (null == child) { return categoryTreeDTO; } categoryTreeDTO.setChildren(child); //组装子类目 for (CategoryTreeDTO dto : child) { assembleTree(dto, allList,remainRecursionCount); } return categoryTreeDTO; }
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

小廖IT

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值