组织结构树处理-告别递归-CSDN博客

本文链接：https://blog.csdn.net/weixin_44345114/article/details/128259207

基本现状

一直以来，我们对于组织结构树的处理，第一印象就是递归，因为谁也不知道到底有多少层。但是当组织结构的叶子变得越来越大，还有比较恶心的场景非要返回整棵树怎么办，还是感觉能跑就行？没到stack overflow的地步就不处理？还是说把锅甩给前端，返回个List让前端开发人员自己处理（虽说不太厚道，但不失为一个好办法。哈哈）这样场景，递归的时间复杂度为O(2^n)，指数型增长，效率指数型下降，滋滋~。

解决办法

怎样效率又高又不会出现栈溢出的情况呢，我的思路是利用Map。我们把查出来的组织列表转成一个Map用ID作为key，自身作为value。那么可能就有人有疑问了，用ID做key怎么查parentId等于自己的下属组织呢（那个…汽车有R档，可以倒着开）。言归正传，我们开始遍历组织列表的list，每次拿到值之后，我们就当他是最底层的节点，要做的就是从Map中寻找他的parentId。然后将自己塞到map中父节点的children下。对，就是在Map中通过不断的引用来形成我们需要的树。当一轮遍历处理完之后，map中就会形成大大小小的树。我们只需要吧根节点的树拿出来就可以。
上代码。一层循环搞定O(n)线程增长。逻辑代码简洁，通俗易懂。完美

private OrgTreeResDTO buildOrgTree(List<Org> orgList,Integer orgId) {
		List<OrgTreeResDTO> orgTreeList = orgList.stream().map(OrgAssembler::to).collect(Collectors.toList());
		Map<Integer, OrgTreeResDTO> orgMap = orgTreeList.stream().collect(Collectors.toMap(OrgTreeResDTO::getOrgId, Function.identity()));
		for (OrgTreeResDTO org : orgTreeList) {
			OrgTreeResDTO parentOrg = orgMap.get(org.getParentId());
			if (Objects.nonNull(parentOrg)) {
				parentOrg.getChildren().add(org);
			}
		}
		return orgMap.get(orgId);
	}