Java List结构转Tree树形结构_非递归_简单优化版

背景

之前写过一个文章 Java递归将List数据转换为Tree结构数据_填坑指南 用的是递归的方法，后来在今日头条上也发布了，评论区以为大佬给了我一些指点（如下图），我看了之后恍然大悟，相对于使用递归，此方法只需写一个方法，方法中两个循环，而使用递归的话，需要写三个方法，还要用递归，使用递归不仅内存开销大，性能也不行。

• 大佬提供的代码

具体实现

    /**
     * 对象List转为Tree树形结构
     * @param entityList 传进来的泛型List
     * @param parentFieldName 父级字段名称
     * @return
     */
    public final List<Map<String, Object>> listToTree1(List<T> entityList, String parentFieldName){
        //返回的map Tree树形结构
        List<Map<String, Object>> treeMap = new ArrayList<>();
        //将传进的参数entityList转为MapList
        List<Map<String, Object>> listMap = JSON.parseObject(JSON.toJSONString(entityList), List.class);
        //声明一个map用来存listMap中的对象，key为对象id，value为对象本身
        Map<String, Map<String, Object>> entityMap = new Hashtable<>();
        //循环listMap把map对象put到entityMap中去
        listMap.forEach(map -> entityMap.put(map.get("id").toString(), map));
        //循环listMap进行Tree树形结构组装
        listMap.forEach(map -> {
            //获取map的pid
            Object pid = map.get(parentFieldName);
            if (pid == null){ //判断pid是否为空，为空说明是最顶级，直接add到返回的treeMap中去
                treeMap.add(map);
            } else { //如果pid不为空
                //根据当前map的pid获取上级 parentMap
                Map<String, Object> parentMap = entityMap.get(pid);
                if (parentMap == null){ //如果parentMap为空，则说明当前map没有父级，当前map就是顶级
                    treeMap.add(map);
                } else {    //如果parentMap不为空，则当前map为parentMap的子级
                    //取出parentMap的所有子级的List集合
                    List<Map<String, Object>> children = (List<Map<String, Object>>)parentMap.get("children");
                    if (children == null){  //判断子级集合是否为空，为空则新创建List
                        children = new ArrayList<>();
                        parentMap.put("children", children);
                    }
                    //把当前map对象add到parentMap的子级List中去
                    children.add(map);
                    /**
                     * 因为parentMap是从entityMap中get出来的，
                     * 而entityMap中的value又是来自于listMap对象，
                     * 所以parentMap和entityMap中的value的地址都是指向listMap中的对象，
                     * 所以parentMap的children和entityMap中的value的children改变时，都会改变listMap中的对象，
                     * 这里涉及到了地址、指针，就不多说了。
                     */
                }
            }
        });
        return treeMap;
    }

总结

因为parentMap是从entityMap中get出来的，而entityMap中的value又是来自于listMap对象，所以parentMap和entityMap中的value的地址都是指向listMap中的对象，所以parentMap的children和entityMap中的value的children改变时，都会改变listMap中的对象，这里涉及到了地址、指针，就不多说了。
这里再次感谢今日头条【 托尼老师学编程 】提供的思路，大家有什么可以优化的想法请在评论区留言，谢谢。

更多内容请点击【阅读原文】