基本现状
一直以来,我们对于组织结构树的处理,第一印象就是递归,因为谁也不知道到底有多少层。但是当组织结构的叶子变得越来越大,还有比较恶心的场景非要返回整棵树怎么办,还是感觉能跑就行?没到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);
}
2022年了,还有java程序员认为学算法没有用吗?一直靠着第三方的jar和工作年限称霸一方吗?哈哈~嗝 学而不思则罔,思而不学则殆。祝愿看到这篇文章又有能力的程序员,不要被学历而卡着职业生涯的咽喉!