java遍历list生成树_java 快速生成树的方式

public class XzqhDto {

@ApiModelProperty("另加数据")

private String label;

@ApiModelProperty("另加数据")

private String value;

@ApiModelProperty("当前本身数据")

private Xzqh xzqh;

@ApiModelProperty("子元素")

List children = new ArrayList<>();

}

/**

* 处理思路

* 1、首先按照顺序查找所有的元素

* 2、转成map形式 提取所有的父类Map1 所有子类map2

* 3、遍历map2 根据key 值塞入 map1中

*

* @return

*/

public Object queryAllAboutXcx() {

//查询所以的数据

List xzqhList = xzqhMapper.queryXzqh(null, null);

// TreeMap 能保证顺序相同

Map map = new TreeMap<>();

Map result = new TreeMap<>();

//转成所有的map

for (Xzqh xzqh : xzqhList) {

XzqhDto xzqhDto = new XzqhDto().setXzqh(xzqh).setLabel(xzqh.getXzqhdmCd()).setValue(xzqh.getXzqhmcNm());

if (xzqh.getSjxzqhdm().length() == 1) {

result.put(xzqh.getXzqhdmCd(), xzqhDto);

} else {

map.put(xzqh.getXzqhdmCd(), xzqhDto);

}

}

//遍历map2

for (Map.Entry entry : map.entrySet()) {

String key = entry.getKey();

if (key.length() == 4) {

String k = key.substring(0, 2);

List xzqhs = result.get(k).getChildren();

xzqhs.add(entry.getValue());

result.get(k).setChildren(xzqhs);

} else if (key.length() == 6) {

String k = key.substring(0, 4);

List xzqhs = map.get(k).getChildren();

xzqhs.add(entry.getValue());

map.get(k).setChildren(xzqhs);

}

}

return result;

}