如果在节点的属性中,增加一个level属性,即树的深度,构造树会非常容易。前提是,增加和修改节点的时候,要维护level。
根节点的level为1,下一级为2,以此类推。
构造树的方法,主要有2个:
// 按照level排序,根节点在上,子结点在下
public static List<Map<String, Object>> buildTree(List<TreeNode> list) {
List<Map<String, Object>> rootList = new ArrayList<Map<String, Object>>();
Map<String, Map<String, Object>> rootMap = new HashMap<String, Map<String, Object>>();
for (TreeNode node : list) {
Integer acl = node.getAcl();
Integer parentAcl = node.getParentAcl();
Map<String, Object> newNode = createNode(acl);
rootMap.put(acl + "", newNode);
if (parentAcl.equals(-1)) {
rootList.add(newNode);
} else {
Map<String, Object> fatherNode = rootMap.get(parentAcl + "");
addChild(fatherNode, newNode);
}
}
return rootList;
}
private static void addChild(Map<String, Object> father,
Map<String, Object> child) {
//如果father=null,一定是没有正确按照level排序导致的
if (father == null) {
System.out.println("error,Father is null");
return;
}
Object o = father.get("children");
List<Map<String, Object>> childs;
if (o == null) {
childs = new ArrayList<Map<String, Object>>();
father.put("children", childs);
} else {
childs = (List<Map<String, Object>>) o;
}
childs.add(child);
}
只需要简单说几点:
1.按照level升序,顶级节点在最前面。
2.把每一个节点的父结点,先放进map。轮到子结点的时候,把它放进父结点的children中。
3.integer比较的时候, 用queals方法,而不是==。
4.Map的key是区分类型的。
map.put(1,"a");
map.get("1");
放进去的值,和取出来的值很可能不一样。
map的key不是同一个,integer的1和string的“1”是不同的。
5.addChild中的父结点,不可能为null。如果出现,一定是没有按照level正确排序。
这种通过level的方式,构造树,好处是,构造很简单,坏处是,还得维护level,麻烦。