Java递归把List建立为树形结构

这段代码主要展示了如何构建树形结构,包括获取根节点和递归建立子树的方法。同时提供了两种模糊查询策略,一种是基于名称的精确匹配,另一种是支持菜单和名称查询,并在下拉中展示。在查询过程中,如果节点不包含关键字,则会从树结构中移除其及下级节点。

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

/**
     * 建立树形结构
     *
     * @param menuList
     * @return
     */
    private List<PpmProjectDto> buildTree(List<PpmProjectDto> menuList) {
        List<PpmProjectDto> treeMenus = new ArrayList<PpmProjectDto>();
        for (PpmProjectDto menuNode : getRootNode(menuList)) {
            buildChildTree(menuList, menuNode);
            treeMenus.add(menuNode);
        }
        return treeMenus;
    }

    /**
     * 获取根节点
     *
     * @param menuList
     * @return
     */
    private List<PpmProjectDto> getRootNode(List<PpmProjectDto> menuList) {
        List<PpmProjectDto> rootMenuLists = new ArrayList<PpmProjectDto>();
        for (PpmProjectDto menuNode : menuList) {
            if (menuNode.getParentId().equals(-1L)) {
                rootMenuLists.add(menuNode);
            }
        }
        return rootMenuLists;
    }

    /**
     * 递归,建立子树形结构
     *
     * @param menuList
     * @param pNode
     * @return
     */
    private PpmProjectDto buildChildTree(List<PpmProjectDto> menuList, PpmProjectDto pNode) {
        List<PpmProjectDto> childMenus = new ArrayList<PpmProjectDto>();
        for (PpmProjectDto menuNode : menuList) {
            if (menuNode.getParentId().equals(pNode.getProjectId())) {
                childMenus.add(buildChildTree(menuList, menuNode));
            }
        }
        pNode.setChildren(childMenus);
        return pNode;
    }

再通过树结构去模糊查询

第一种:查询精确的某个名称

    /**
     * 再通过树结构去模糊查询
     *
     * @param data
     * @param keyword
     * @return
     */
    private List<TreeDto> filterTreeByKeyWord(List<TreeDto> data, String keyword) {
        if (CollectionUtils.isEmpty(data)) {
            return data;
        }
        filter(data, keyword);
        return data;
    }

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

第二种:支持查询菜单和名字,查询菜单时支持下拉

/**
     * 再通过树结构去模糊查询
     *
     * @param data
     * @param keyword
     * @return
     */
    private List<PpmProjectDto> filterTreeByName(List<PpmProjectDto> data, String keyword) {
        if (CollectionUtils.isEmpty(data)) {
            return data;
        }
        filter(data, keyword, null);
        return data;
    }

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

DevanLove

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

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

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

打赏作者

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

抵扣说明:

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

余额充值