记一个递归封装树形结构

最近改了个小bug,原有的数据字典查询,封装成树的递归写的有问题,就自己写了一个,解决这种递归的思考是:分析树形结构–>>找到父子层之间的关联关系–>>根据关系编写递归嵌套条件。代码很简单也很实用,遂总结如下

表结构

截取表结构如下,关联id 和pid(父id).其中树最外层pid为0
记一个递归封装树形结构_第1张图片
代码逻辑

public List<Map<String, Object>> getRootDict() {
	List<Map<String, Object>> returnList = new ArrayList<>();
	// 查询所有数据字典
	List<Map<String, Object>> dictList = dictMapper.getAllDict();
	Map<String, Object> map = new HashMap<>(16);
	map.put("name", "数据字典");
	map.put("id", null);
	map.put("pid", null);
	
	// 构建树形结构
	List<Map<String, Object>> rootTree = this.rebuildRootTree("0",dictList);
	map.put("child", rootTree);
	returnList.add(map);
	return returnList;
}

/**
 * 构建树结构
 * @param pid
 * @param dictList
 * @return
 */
List<Map<String, Object>> rebuildRootTree(String pid ,List<Map<String, Object>> dictList){
	List<Map<String, Object>> rootTree = new ArrayList<>();
	for(Map<String, Object> dictMap:dictList){
		if(dictMap.containsKey("pid") && Objects.equals(dictMap.get("pid").toString(),pid)) {
			Map<String, Object> newMap = new HashMap<>(16);
			newMap.put("name", dictMap.get("name"));
			newMap.put("id", dictMap.get("id"));
			newMap.put("pid", dictMap.get("pid"));
			//递归调用
			newMap.put("child",this.rebuildRootTree((dictMap.get("id")).toString(),dictList));
			rootTree.add(newMap);
		}
	}
	return rootTree;
}

若为实体,替换为对应的实体类即可;好了,就酱!日常生活的小技巧,还是总结下

你可能感兴趣的:(后端,java)