java 递归遍历树形结构

本文介绍了一种构建菜单树形结构的方法,通过递归遍历菜单列表并根据父节点ID来组装子节点,最终形成树状结构。示例中包含了不同层级的菜单项。
好记性不如烂笔头,记录下。

//菜单树形结构  
   public JSONArray treeMenuList(JSONArray menuList, int parentId) {  
       JSONArray childMenu = new JSONArray();  
       for (Object object : menuList) {  
           JSONObject jsonMenu = JSONObject.fromObject(object);  
           int menuId = jsonMenu.getInt("id");  
           int pid = jsonMenu.getInt("parentId");  
           if (parentId == pid) {  
               JSONArray c_node = treeMenuList(menuList, menuId);  
               jsonMenu.put("childNode", c_node);  
               childMenu.add(jsonMenu);  
           }  
       }  
       return childMenu;  
   }  

public static void main(String args[]) {  
       MenuCacheService menuCacheService = new MenuCacheService();  
       JSONArray jsonArray = new JSONArray();  
       Menu menu1 = new Menu();  
       menu1.setId(1l);  
       menu1.setParentId(0);  
       menu1.setLevel(0);  
       Menu menu2 = new Menu();  
       menu2.setId(2l);  
       menu2.setParentId(0);  
       menu2.setLevel(0);  
       Menu menu3 = new Menu();  
       menu3.setId(3l);  
       menu3.setParentId(2);  
       menu3.setLevel(1);  
       Menu menu4 = new Menu();  
       menu4.setId(4l);  
       menu4.setParentId(2);  
       menu4.setLevel(1);  
       Menu menu5 = new Menu();  
       menu5.setId(5l);  
       menu5.setParentId(4);  
       menu5.setLevel(2);  
       Menu menu6 = new Menu();  
       menu6.setId(6l);  
       menu6.setParentId(1);  
       menu6.setLevel(1);  

       jsonArray.add(menu1);  
       jsonArray.add(menu2);  
       jsonArray.add(menu3);  
       jsonArray.add(menu4);  
       jsonArray.add(menu5);  
       jsonArray.add(menu6);  

       System.out.print(menuCacheService.treeMenuList(jsonArray,0));  
   }  
/** * 根据等级查询类目树 * * @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; }
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值