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

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

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

主键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

遍历的总次数：866674=？ //其实也挺多的次数。。

分享一：

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

主键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这样的数据，叫什么名字都无所谓，不管多少层，都可以给弄出来。觉得这个特别好了。感谢@小太阳の ！！
   经过使用，发现，该方法，在很多的时候，特别慢。最后又找到一个好方法。

分享二：（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);
    }
}

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

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