java树结构模糊查询

//原始数据先转成树结构

 private List<TaobaoCategoryResp> recursionMethodTaoBao(List<TaobaoCategoryResp> treeList) {
        List<TaobaoCategoryResp> trees = new ArrayList<>();
        for (TaobaoCategoryResp tree : treeList) {
            // 找出父节点
            if ("0".equals(tree.getParentCategoryId())) {
                // 调用递归方法填充子节点列表
                trees.add(findChildrenTaoBao(tree, treeList));
            }
        }
        return trees;
    }

    private TaobaoCategoryResp findChildrenTaoBao(TaobaoCategoryResp tree, List<TaobaoCategoryResp> treeList) {
        for (TaobaoCategoryResp node : treeList) {
            if (tree.getCategoryId().equals(node.getParentCategoryId())) {
                if (tree.getChildren() == null) {
                    tree.setChildren(new ArrayList<>());
                }
                // 递归 调用自身
                tree.getChildren().add(findChildrenTaoBao(node, treeList));
            }
        }
        return tree;
    }


//再通过树结构去模糊查询
 private List<TaobaoCategoryResp> filterTreeByKeyWordByTaoBao(List<TaobaoCategoryResp> data, String keyword) {

        if (CollectionUtils.isEmpty(data)) {
            return data;
        }
        filterByTaoBao(data, keyword);
        return data;
    }

    /**
     * 递归方法
     * @param list 任意层级的节点
     * @param keyword 关键字
     */
    private void filterByTaoBao(List<TaobaoCategoryResp> list, String keyword){
        Iterator<TaobaoCategoryResp> parent = list.iterator();
        while (parent.hasNext()){
            //当前节点
            TaobaoCategoryResp t = parent.next();
            if (StrUtil.isNotEmpty(keyword) && !t.getCategoryName().contains(keyword)) {
                //当前节点不包含关键字,继续遍历下一级
                // 取出下一级节点
                List<TaobaoCategoryResp> children = t.getChildren();
                // 递归
                if (!CollectionUtils.isEmpty(children)) {
                    filterByTaoBao(children, keyword);
                }
                //下一级节点都被移除了,那么父节点也移除,因为父节点也不包含关键字
                if (CollectionUtils.isEmpty(t.getChildren())) {
                    parent.remove();
                }
            } else {
                //当前节点包含关键字,继续递归遍历
                //子节点递归如果不包含关键字则会进入if分支被删除
                List<TaobaoCategoryResp> children = t.getChildren();
                // 递归
                if (!CollectionUtils.isEmpty(children)) {
                    filterByTaoBao(children, keyword);
                }
            }
        }
    }





  • 9
    点赞
  • 28
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
/** * 根据等级查询类目树 * * @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; }
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值