我们在展示一个机构树的时候,常常会遇到这样的一个问题,查询数据的时候,是从下往上查的,但展示数据的时候,又要从下往上展示。
这时候就要把查询到的数据进行整理从而得到我们想要的结构。
举个例子。
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的童鞋的批评。