无限层级且乱序的树形结构数据的整理,利用HashMap降低遍历次数
我们在展示一个机构树的时候,经常会遇到这种一个问题,查询数据的时候,是从下往上查的,但展示数据的时候,又要从下往上展示。
这时候就要把查询到的数据进行整理从而得到我们想要的结构。
举个样例。
ID | PARENT_ID | SOME_ATTRIBUTE_ID |
2001 | 0 | |
6292 | 6120 | 57010 |
6120 | 6115 | |
6121 | 6115 | |
6156 | 6121 | 56874 |
6115 | 2001 |
这是依据需求查询出的sql数据。可是它是无序的,所以非常让人头疼,不知怎样去处理,示意图是这种。
我们先明白下数据结构吧。
每个节点我们使用一个Map存储内容,key-value映射例如以下。
key | value |
ID | String |
Parent_ID | String |
Attribute_id | String |
Children | List<Map> |
同一时候须要说明的是,我们的原始数据就是一个乱序的List<Map>,map中包括前三项内容。
最简单的办法就是有几层就遍历几次List。第一次遍历整个List,查找PID为0的节点,新建空的List放入Map中。这次遍历我们拿到2001这个节点,并把这个节点从List中清除。
第二次遍历,查找PID为2001的节点,这次我们查到6115这个节点。依次类推。遍历四次。我们就依照层次结构形成了须要的数据。
可是这样效率不好,有没有办法能遍历一次就完毕数据的整理工作呢?
看我把代码贴出来:
<span style="white-space:pre"> </span>List<Map> list = getList(); Map all = new HashMap(); for(int i = 0;i<list.size();i++){ Map result = list.get(i); String parent_id = (String) result.get("PARENT_ID"); String id = (String) result.get("ID"); if(all.get(parent_id) == null){ Map temp = new HashMap(); List tempList = new ArrayList(); if(all.get(id) != null){ ((Map)all.get(id)).putAll(result); }else{ result.put("children", new ArrayList()); all.put(id, result); } tempList.add(all.get(id)); temp.put("children", tempList); all.put(parent_id, temp); }else{ if(all.get(id) == null){ result.put("children", new ArrayList()); all.put(id, result); }else{ ((Map)all.get(id)).putAll(result); } ((List)((Map)all.get(parent_process)).get("children")).add(result); } }
少了遍历,就要多加入逻辑。
list是我们查询的内容,我们遍历list的时候,每拿到一条。就查看在all中。是否已经存在key为parent_id的对象,假设没有,我们再看有没有key为id的对象,假设有。它一定是作为parent_id时被建立的,所以我们把其它的内容加进去,假设没有,我们就在现有的result基础上,加入key为children的list,并把它以id为key保存在all中。同一时候。再新建一个以parent_id为key的对象,当中包括children为key的List。里边包括了key为id的对象。
假设以parent_id为key的对象在all中存在。那就简单了,仅仅须要查看all中是否有以id为key的对象,有直接将其加入至parent_id为key的对象中。没有的话。建立它,再把它放进去。
逻辑上确实比較复杂,也不是非常好理解。总之我们是在all中保存了全部以不论什么一个节点为顶节点的树,仅仅须要遍历一遍,整个all中的东西都能被整理完毕。
每次做到类似的问题的时候,都非常懊悔上大学的时候对acm嗤之以鼻。
事实上如今还是有点嗤之以鼻。。。。
我认为这根本不叫算法啊。数学模型才叫算法啊。
。。。
求醍醐灌顶!
另外本文求更优的解法。尤其是学过acm的童鞋的批评。