JAVA代码实现多级树结构封装对象

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

以地区（区、镇、村）为例

后台一般对于树结构对象在数据库的结构是这样的：

主键ID	名字	父ID
ID	REGION_NAME	PARENT_ID
121100	尼龙区	0
121100001	尼龙区钢丝镇	121100
121100001001	尼龙区钢丝镇螺丝村	121100001

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

主键ID	名字	词典表类型
ID	REGION_NAME	DICT_TYPE
121100	尼龙区	XZQH
121100001	尼龙区钢丝镇	XZQH
121100001001	尼龙区钢丝镇螺丝村	XZQH
231111	一般性支付	ZFFS

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

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

首先根据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

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

分享一：

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

主键ID	名字	父ID
ID	REGION_NAME	PARENT_ID
121100	尼龙区	0
121100001	尼龙区钢丝镇	121100
121100001001	尼龙区钢丝镇螺丝村	121100001

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

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

赋上他给我的两个工具类：

工具类下载地址

使用的主要逻辑在TreeUtils类里，在这里边我加了一个main方法，模拟了一个例子，大家可以参考。最后实现的结果如下图：



他的这个工具类的好处是，利用了反射，只要你的数据结构符合上边的id、pid、name这样的数据，叫什么名字都无所谓，不管多少层，都可以给弄出来。觉得这个特别好了。感谢@小太阳の ！！

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