java的List列表轉成Tree（樹形）結構列表

直接看借鑑博客：https://blog.csdn.net/massivestars/article/details/53911620/

 

由於我的業務沒有父子級id兩個字段，只有一個層級id字段來分層，如一級數據：01，02，03...，二級數據：0101，0102，0103，0201，...，三級數據：010101，010102...等等，但思路是一樣的，我用的是兩個for循環搞定，如下：

/**
     * @Description：list轉tree（樹形方法）
     * @Author：
     * @Date：2018/12/16 16:19
     **/
    public static List<WkgGroupVo> listToTree(List<WkgGroupVo> wkgGroupVoList){
        List<WkgGroupVo> wkgGroupVoTree = new ArrayList<WkgGroupVo>();
        for(WkgGroupVo wgVo : wkgGroupVoList){

            //層級字段數據爲兩位數，則是一級數據
            if(wgVo.getGpStructurecode().length() == 2){
                wkgGroupVoTree.add(wgVo);
            }

            //循環出子級數據，添加到一級數據
            for(WkgGroupVo subWgVo : wkgGroupVoList){
                int len = subWgVo.getGpStructurecode().length();
                String gpStructurecode = subWgVo.getGpStructurecode().substring(0, len -2);
                if(gpStructurecode.equals(wgVo.getGpStructurecode())){
                    if(wgVo.getWgList() == null){
                        wgVo.setWgList(new ArrayList<WkgGroupVo>());
                    }
                    wgVo.getWgList().add(subWgVo);
                }
            }

        }

        return wkgGroupVoTree;
    }