我们经常在代码里会造一个树结构对象,以方便前端使用。
以地区(区、镇、村)为例
后台一般对于树结构对象在数据库的结构是这样的:
主键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、建立一个对象,用来规范树结构。
<span style="color:#000000"><code><span style="color:#000088">public</span> <span style="color:#000088">class</span> <span style="color:#4f4f4f">FpzjXzdwArray</span> <span style="color:#000088">implements</span> <span style="color:#4f4f4f">Serializable</span> {
<span style="color:#000088">private</span> String value; <span style="color:#880000">//对应数据库的ID</span>
<span style="color:#000088">private</span> String text; <span style="color:#880000">//对应数据库的region_name</span>
<span style="color:#000088">private</span> List<FpzjXzdwArray> children;
<span style="color:#880000">//省略get/set方法了</span>
}</code></span>
- 1
- 2
- 3
- 4
- 5
- 6
2、第一步我就省略了。
3、第二步像我这种情况就是根据ID了,长度为6的放到区map里,长度为9的放到镇map里,长度为12的放到村map里。
<span style="color:#000000"><code><span style="color:#000088">Map</span><<span style="color:#4f4f4f">String</span>,FpzjXzdwArray> quMap = <span style="color:#000088">new</span> HashMap<<span style="color:#4f4f4f">String</span>,FpzjXzdwArray>();
<span style="color:#000088">Map</span><<span style="color:#4f4f4f">String</span>,FpzjXzdwArray> zhenMap = <span style="color:#000088">new</span> HashMap<<span style="color:#4f4f4f">String</span>,FpzjXzdwArray>();
<span style="color:#000088">Map</span><<span style="color:#4f4f4f">String</span>,FpzjXzdwArray> cunMap = <span style="color:#000088">new</span> HashMap<<span style="color:#4f4f4f">String</span>,FpzjXzdwArray>();</code></span>
- 1
- 2
- 3
4、嵌套遍历三个map,首先缕缕我们的步骤。
以一个小例子为例,我们现在有一个大箱子、2个中箱子(1红、1蓝)、6个红蓝小箱子(3红、3蓝)。要求是把大箱子里装上中箱子,中箱子里要颜色对应装进去小箱子。
正确逻辑:如果让一个正常人来做,我们的逻辑都是一样的,要先把颜色对应的小箱子放到中箱子里以后,才会把中箱子放到大箱子里,对吗?
错误逻辑:如果你非要先把中箱子放到大箱子以后,再去拿对应颜色的小箱子放进中箱子里去的话,这样带来的一个问题就是,你必须要先把大箱子打开才可以完成这种事。所以这种逻辑就是错误的。
这个例子反映到代码里也是一样,如果你先往区对象里塞进镇对象集合,然后再去往镇对象里塞对应的村对象,就会像上边的第二个步骤一样,逻辑就错了。我的代码如下:(最下边还有别人分享给我的好方法,先贴上我的,瞬间感觉自己弱爆了)
<span style="color:#000000"><code><span style="color:#000088">for</span>(String qu:quMap.keySet()){ <span style="color:#880000">//遍历所有的区</span>
<span style="color:#000088">for</span>(String zhen:zhenMap.keySet()){ <span style="color:#880000">//遍历所有的镇</span>
<span style="color:#000088">if</span>(qu.equals(zhen.substring(<span style="color:#006666">0</span>,<span style="color:#006666">6</span>))){ <span style="color:#880000">//该区下找到该镇</span>
FpzjXzdwArray zhenXzdw = <span style="color:#000088">null</span>;<span style="color:#880000">//定义这个是有必要的,这样可以先完成对村对象塞到镇对象里,然后再完成镇对象塞到区对象的步骤。</span>
<span style="color:#000088">for</span>(String cun:cunMap.keySet()){ <span style="color:#880000">//有区有镇,然后再找下一级村</span>
<span style="color:#000088">if</span>(zhen.equals(cun.substring(<span style="color:#006666">0</span>,<span style="color:#006666">9</span>))){ <span style="color:#880000">//该镇下找到村</span>
zhenXzdw = zhenMap.<span style="color:#000088">get</span>(zhen); <span style="color:#880000">//得到该镇对象,</span>
List<FpzjXzdwArray> xzdwArrayZhenList = zhenXzdw.getChildren(); <span style="color:#880000">//取出镇对象下的村集合</span>
<span style="color:#000088">if</span>(xzdwArrayZhenList == <span style="color:#000088">null</span>){
xzdwArrayZhenList = <span style="color:#000088">new</span> ArrayList<>();
xzdwArrayZhenList.add(cunMap.<span style="color:#000088">get</span>(cun));
zhenXzdw.setChildren(xzdwArrayZhenList); <span style="color:#880000">//将符合该镇下的村塞到镇对象里</span>
}<span style="color:#000088">else</span>{
xzdwArrayZhenList.add(cunMap.<span style="color:#000088">get</span>(cun));
zhenXzdw.setChildren(xzdwArrayZhenList); <span style="color:#880000">//将符合该镇下的村塞到镇对象里</span>
}
}
}
FpzjXzdwArray quXzdw = quMap.<span style="color:#000088">get</span>(qu); <span style="color:#880000">//得到该区对象,</span>
List<FpzjXzdwArray> xzdwArrayList = quXzdw.getChildren();<span style="color:#880000">//取出区对象下的镇集合</span>
<span style="color:#000088">if</span>(xzdwArrayList == <span style="color:#000088">null</span>){
xzdwArrayList = <span style="color:#000088">new</span> ArrayList<>();
xzdwArrayList.add(zhenXzdw);
quXzdw.setChildren(xzdwArrayList); <span style="color:#880000">//将符合该区下的镇对象塞到区对象里</span>
quMap.put(qu,quXzdw);
}<span style="color:#000088">else</span>{
xzdwArrayList.add(zhenXzdw);
quXzdw.setChildren(xzdwArrayList);<span style="color:#880000">//将符合该区下的镇对象塞到区对象里</span>
quMap.put(qu,quXzdw);
}
}
}
}
<span style="color:#000088">return</span> quMap.values();</code></span>
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
以上就是我写的实现的逻辑,我还没有用上递归,本质是这样。
消耗的时间我也测试了,对我来说几毫秒,最多7ms。
我的数据的数量是:
数量 | 区 | 镇 | 村 |
---|---|---|---|
总数 | 8 | 66 | 674 |
遍历的总次数:8*66*674=? //其实也挺多的次数。。
分享一:
在我加的技术群里吆喝了半天,有一位朋友分享他们的工具类,我感觉写的很好了。他需要数据的结构是:
主键ID | 名字 | 父ID |
---|---|---|
ID | REGION_NAME | PARENT_ID |
121100 | 尼龙区 | 0 |
121100001 | 尼龙区钢丝镇 | 121100 |
121100001001 | 尼龙区钢丝镇螺丝村 | 121100001 |
对应封装好的实体类对象可以是这样的:
<span style="color:#000000"><code><span style="color:#000088">public</span> <span style="color:#000088">class</span> TestTree{
<span style="color:#000088">private</span> String id;
<span style="color:#000088">private</span> String name;
<span style="color:#000088">private</span> String pid;
<span style="color:#000088">private</span> List<TestTree> testTrees;
<span style="color:#880000">//省略get/set方法</span>
}</code></span>
- 1
- 2
- 3
- 4
- 5
- 6
- 7
赋上他给我的两个工具类:
工具类下载地址
使用的主要逻辑在TreeUtils类里,在这里边我加了一个main方法,模拟了一个例子,大家可以参考。最后实现的结果如下图:
他的这个工具类的好处是,利用了反射,只要你的数据结构符合上边的id、pid、name这样的数据,叫什么名字都无所谓,不管多少层,都可以给弄出来。觉得这个特别好了。感谢@小太阳の !!
造树我觉得是挺重要的,挺常用的,如果有多级树,我这种方式就不适用,所以我想找到一个可以封装多级树的好代码,希望有能看到,留言共享一下,我再把好的方法补充到这个文章里。