首页 > 代码库 > 无限层级且乱序的树形结构数据的整理,利用HashMap减少遍历次数

无限层级且乱序的树形结构数据的整理,利用HashMap减少遍历次数

我们在展示一个机构树的时候,常常会遇到这样的一个问题,查询数据的时候,是从下往上查的,但展示数据的时候,又要从下往上展示。

这时候就要把查询到的数据进行整理从而得到我们想要的结构。

举个例子。

IDPARENT_IDSOME_ATTRIBUTE_ID
20010 
6292612057010
61206115 
61216115 
6156612156874
61152001 

这是根据需求查询出的sql数据,但是它是无序的,所以很让人头疼,不知如何去处理,示意图是这样的。


我们先明确下数据结构吧。每一个节点我们使用一个Map存储内容,key-value映射如下。

keyvalue
IDString
Parent_IDString
Attribute_idString
ChildrenList<Map>
children用来储存它的子节点的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的童鞋的批评。