JAVA代码实现多级树结构封装对象（2018-09-26补充）

╰半夏微凉° 2024-02-18 16:02 3阅读 0赞

##### **我们经常在代码里会造一个树结构对象，以方便前端使用。** #####

以地区（区、镇、村）为例  
后台一般对于树结构对象在数据库的结构是这样的：

<table> 
 <thead> 
  <tr> 
   <th>主键ID</th> 
   <th align="center">名字</th> 
   <th align="right">父ID</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>ID</td> 
   <td align="center">REGION_NAME</td> 
   <td align="right">PARENT_ID</td> 
  </tr> 
  <tr> 
   <td>121100</td> 
   <td align="center">尼龙区</td> 
   <td align="right">0</td> 
  </tr> 
  <tr> 
   <td>121100001</td> 
   <td align="center">尼龙区钢丝镇</td> 
   <td align="right">121100</td> 
  </tr> 
  <tr> 
   <td>121100001001</td> 
   <td align="center">尼龙区钢丝镇螺丝村</td> 
   <td align="right">121100001</td> 
  </tr> 
 </tbody> 
</table>

其实这样返回给前端，前端是可以做成树的，但是我这里不是要说这种的，我遇到的是下边的情况。

<table> 
 <thead> 
  <tr> 
   <th>主键ID</th> 
   <th align="center">名字</th> 
   <th align="right">词典表类型</th> 
  </tr> 
 </thead> 
 <tbody> 
  <tr> 
   <td>ID</td> 
   <td align="center">REGION_NAME</td> 
   <td align="right">DICT_TYPE</td> 
  </tr> 
  <tr> 
   <td>121100</td> 
   <td align="center">尼龙区</td> 
   <td align="right">XZQH</td> 
  </tr> 
  <tr> 
   <td>121100001</td> 
   <td align="center">尼龙区钢丝镇</td> 
   <td align="right">XZQH</td> 
  </tr> 
  <tr> 
   <td>121100001001</td> 
   <td align="center">尼龙区钢丝镇螺丝村</td> 
   <td align="right">XZQH</td> 
  </tr> 
  <tr> 
   <td>231111</td> 
   <td align="center">一般性支付</td> 
   <td align="right">ZFFS</td> 
  </tr> 
 </tbody> 
</table>

#####   数据库的结构如上，并没有树结构，需要根据DICT\_TYPE为XZQH的记录取出来，然后封装成树结构返回给前端，前端要的树结构类似如下： #####

![这里写图片描述][SouthEast]

##### **我的实现的逻辑是这样的：** #####

*  首先根据DICT\_TYPE取出相关的地区记录。
 *  然后将区、镇、村分别存到三个不同的map里。
 *  最后，嵌套遍历三个map的key，区里边套镇、镇里边套村，封装对象。（最重要的地方）。

##### **具体步骤：** #####

##### 1、建立一个对象，用来规范树结构。 #####

public class FpzjXzdwArray implements Serializable 	{
        private String value; //对应数据库的ID
        private String text; //对应数据库的region_name
        private List<FpzjXzdwArray> children;
        //省略get/set方法了
    }

##### 2、第一步我就省略了。 #####

##### 3、第二步像我这种情况就是根据ID了，长度为6的放到区map里，长度为9的放到镇map里，长度为12的放到村map里。 #####

Map<String,FpzjXzdwArray> quMap = new HashMap<String,FpzjXzdwArray>();
    Map<String,FpzjXzdwArray> zhenMap = new HashMap<String,FpzjXzdwArray>();
    Map<String,FpzjXzdwArray> cunMap = new HashMap<String,FpzjXzdwArray>();

##### 4、嵌套遍历三个map,首先缕缕我们的步骤。 #####

以一个小例子为例，我们现在有一个大箱子、2个中箱子（1红、1蓝）、6个红蓝小箱子（3红、3蓝）。要求是把大箱子里装上中箱子，中箱子里要颜色对应装进去小箱子。  
 \*\*正确逻辑：\*\*如果让一个正常人来做，我们的逻辑都是一样的，要先把颜色对应的小箱子放到中箱子里以后，才会把中箱子放到大箱子里，对吗？  
 \*\*错误逻辑：\*\*如果你非要先把中箱子放到大箱子以后，再去拿对应颜色的小箱子放进中箱子里去的话，这样带来的一个问题就是，你必须要先把大箱子打开才可以完成这种事。所以这种逻辑就是错误的。

#####   这个例子反映到代码里也是一样，如果你先往区对象里塞进镇对象集合，然后再去往镇对象里塞对应的村对象，就会像上边的第二个步骤一样，逻辑就错了。我的代码如下：（**最下边还有别人分享给我的好方法，先贴上我的，瞬间感觉自己弱爆了**） #####

for(String qu:quMap.keySet()){ //遍历所有的区
    			for(String zhen:zhenMap.keySet()){ //遍历所有的镇
    				if(qu.equals(zhen.substring(0,6))){ //该区下找到该镇
    					FpzjXzdwArray zhenXzdw = null;//定义这个是有必要的，这样可以先完成对村对象塞到镇对象里，然后再完成镇对象塞到区对象的步骤。
    					for(String cun:cunMap.keySet()){ //有区有镇，然后再找下一级村
    						if(zhen.equals(cun.substring(0,9))){ //该镇下找到村
    							zhenXzdw = zhenMap.get(zhen); //得到该镇对象，
    							List<FpzjXzdwArray> xzdwArrayZhenList = zhenXzdw.getChildren(); //取出镇对象下的村集合
    							if(xzdwArrayZhenList == null){
    								xzdwArrayZhenList = new ArrayList<>();
    								xzdwArrayZhenList.add(cunMap.get(cun));
    								zhenXzdw.setChildren(xzdwArrayZhenList); //将符合该镇下的村塞到镇对象里
    							}else{
    								xzdwArrayZhenList.add(cunMap.get(cun));
    								zhenXzdw.setChildren(xzdwArrayZhenList); //将符合该镇下的村塞到镇对象里
    							}
    						}
    					}
    					FpzjXzdwArray quXzdw = quMap.get(qu); //得到该区对象，
    					List<FpzjXzdwArray> xzdwArrayList = quXzdw.getChildren();//取出区对象下的镇集合
    					if(xzdwArrayList == null){
    						xzdwArrayList = new ArrayList<>();
    						xzdwArrayList.add(zhenXzdw);
    						quXzdw.setChildren(xzdwArrayList); //将符合该区下的镇对象塞到区对象里
    						quMap.put(qu,quXzdw);
    					}else{
    						xzdwArrayList.add(zhenXzdw);
    						quXzdw.setChildren(xzdwArrayList);//将符合该区下的镇对象塞到区对象里
    						quMap.put(qu,quXzdw);
    					}
    				}
    			}
    		}
    		return quMap.values();

以上就是我写的实现的逻辑，我还没有用上递归，本质是这样。  
消耗的时间我也测试了，对我来说几毫秒，最多7ms。  
我的数据的数量是：

遍历的总次数：8*66*674=？ //其实也挺多的次数。。

##### **分享一：** #####

在我加的技术群里吆喝了半天，有一位朋友分享他们的工具类，我感觉写的很好了。他需要数据的结构是：

对应封装好的实体类对象可以是这样的：

public class TestTree{
        private String id;
        private String name;
        private String pid;
        private List<TestTree> testTrees;
        //省略get/set方法
     }

赋上他给我的两个工具类：  
[工具类下载地址][Link 1]  
使用的主要逻辑在TreeUtils类里，在这里边我加了一个main方法，模拟了一个例子，大家可以参考。最后实现的结果如下图：

![这里写图片描述][SouthEast 1]

他的这个工具类的好处是，利用了反射，只要你的数据结构符合上边的id、pid、name这样的数据，叫什么名字都无所谓，不管多少层，都可以给弄出来。觉得这个特别好了。感谢@小太阳の ！！  
   经过使用，发现，该方法，在很多的时候，特别慢。最后又找到一个好方法。

##### **分享二：（2018-09-26补充）** #####

发现了一个写法超级牛掰的多级树，我修改了一下，代码如下：

package znxd.tjzb.util;
    
    import com.google.common.collect.Lists;
    import com.google.common.collect.Maps;
    import znxd.tjzb.model.region.RegionBeanTree;
    import java.util.ArrayList;
    import java.util.List;
    import java.util.Map;
    
    /**
     * Created by Administrator on 2018/9/26.
     */
    public class TreeToolUtils {
    
        private List<RegionBeanTree> rootList; //根节点对象存放到这里
    
        private List<RegionBeanTree> bodyList; //其他节点存放到这里，可以包含根节点
    
        public TreeToolUtils(List<RegionBeanTree> rootList, List<RegionBeanTree> bodyList) {
            this.rootList = rootList;
            this.bodyList = bodyList;
        }
    
        public List<RegionBeanTree> getTree(){   //调用的方法入口
            if(bodyList != null && !bodyList.isEmpty()){
            //声明一个map，用来过滤已操作过的数据
                Map<String,String> map = Maps.newHashMapWithExpectedSize(bodyList.size());
                rootList.forEach(beanTree -> getChild(beanTree,map));
                return rootList;
            }
            return null;
        }
    
        public void getChild(RegionBeanTree beanTree,Map<String,String> map){
            List<RegionBeanTree> childList = Lists.newArrayList();
            bodyList.stream()
                    .filter(c -> !map.containsKey(c.getCode()))
                    .filter(c ->c.getPid().equals(beanTree.getCode()))
                    .forEach(c ->{
                        map.put(c.getCode(),c.getPid());
                        getChild(c,map);
                        childList.add(c);
                    });
            beanTree.setChildren(childList);
        }
    
        public static void main(String[] args){
            RegionBeanTree beanTree1 = new RegionBeanTree();
            beanTree1.setCode("540000");
            beanTree1.setLabel("西藏省");
            beanTree1.setPid("100000"); //最高节点
            RegionBeanTree beanTree2 = new RegionBeanTree();
            beanTree2.setCode("540100");
            beanTree2.setLabel("拉萨市");
            beanTree2.setPid("540000");
            RegionBeanTree beanTree3 = new RegionBeanTree();
            beanTree3.setCode("540300");
            beanTree3.setLabel("昌都市");
            beanTree3.setPid("540000");
            RegionBeanTree beanTree4 = new RegionBeanTree();
            beanTree4.setCode("540121");
            beanTree4.setLabel("林周县");
            beanTree4.setPid("540100");
            RegionBeanTree beanTree5 = new RegionBeanTree();
            beanTree5.setCode("540121206");
            beanTree5.setLabel("阿朗乡");
            beanTree5.setPid("540121");
            RegionBeanTree beanTree6 = new RegionBeanTree();
            List<RegionBeanTree> rootList = new ArrayList<>();
            rootList.add(beanTree1);
            List<RegionBeanTree> bodyList = new ArrayList<>();
            bodyList.add(beanTree1);
            bodyList.add(beanTree2);
            bodyList.add(beanTree3);
            bodyList.add(beanTree4);
            bodyList.add(beanTree5);
            TreeToolUtils utils =  new TreeToolUtils(rootList,bodyList);
            List<RegionBeanTree> result =  utils.getTree();
            result.get(0);
        }
    }

说实话，这种写法我反正写不出来，只能看懂，汗~，例子我放到最后了，大家可以下载下来直接运行即可。速度杠杠的。

##### **造树我觉得是挺重要的，挺常用的，如果有多级树，我这种方式就不适用，所以我想找到一个可以封装多级树的好代码，希望有能看到，留言共享一下，我再把好的方法补充到这个文章里。** #####

[SouthEast]: https://image.dandelioncloud.cn/pgy_files/images/2024/01/29/773930af1bb94287a791d18eb40875af.png
[Link 1]: http://pan.baidu.com/s/1o78n6ga
[SouthEast 1]: https://image.dandelioncloud.cn/pgy_files/images/2024/01/29/1c9d1c50fd2140b3bff6f58f4e1d669f.png