最近在做菜单管理的展示时,发现树形列表必须按照父在子前且子紧随父才能正确的展示,经过查阅资料后找到两种解决方案,一种是使用递归进行排序,这种方式请参考我的另一篇文章《关于菜单树排序问题的解决方案(一)》;另一种是使用多叉树进行遍历,本文讨论使用多叉树遍历的方案。我对这两种方案都简单封装了下,本文作为备忘。
使用多叉树对菜单树进行遍历
关于树节点接口和具体节点的实体请参考《关于菜单树排序问题的解决方案(一)》,我这里就不再重复了。
- 多叉树
public class MultiTree<N extends TreeNode>{
private N node;
private List<MultiTree> children;
/**
* 返回排好序的List
*/
public List<N> getSortTreeList(List<N> nodeList){
return traverseTree(createTree(nodeList));
}
/**
* 根据菜单树List创建多叉树
*/
public List<MultiTree> createTree(List<N> nList){
List<MultiTree> operateTree=new ArrayList<>();
List<MultiTree> rootTree=new ArrayList<>();
for (N n:nList){
operateTree.add(new MultiTree(n));
}
for (MultiTree tree1:operateTree){
if (tree1.node.isRootNode()){
rootTree.add(tree1);
}
for (MultiTree tree2:operateTree){
if (tree2.node.getPId()!=null&&tree2.node.getPId().equals(tree1.node.getId())){
tree1.children.add(tree2);
}
}
}
return rootTree;
}
/**
* 遍历多叉树并返回排好序的List
*/
public List<N> traverseTree(List<MultiTree> trees){
List<N> resultTree=new ArrayList<>();
for (MultiTree tree:trees){
resultTree.add((N) tree.node);
if (tree.children!=null&&tree.children.size()>0){
resultTree.addAll(traverseTree(tree.children));
}
}
return resultTree;
}
public MultiTree() {
}
public MultiTree(N node) {
this.node = node;
children=new ArrayList<>();
}
}
- 多叉树方式的使用
public class Main {
public static void main(String []args){
List<TestNode> testNodes=new ArrayList<>();
testNodes.add(new TestNode(1,"1",0));
testNodes.add(new TestNode(2,"2",0));
testNodes.add(new TestNode(3,"3",0));
testNodes.add(new TestNode(4,"1-4",1));
testNodes.add(new TestNode(5,"2-5",2));
testNodes.add(new TestNode(6,"3-6",3));
testNodes.add(new TestNode(7,"1-4-7",4));
testNodes.add(new TestNode(8,"1-4-8",4));
testNodes.add(new TestNode(9,"2-5-9",5));
testNodes.add(new TestNode(10,"2-5-9-10",9));
testNodes.add(new TestNode(11,"2-5-9-10-11",10));
testNodes.add(new TestNode(12,"2-5-9-10-11-12",11));
testNodes.add(new TestNode(13,"2-5-9-10-11-12-13",12));
testNodes.add(new TestNode(14,"2-5-9-10-11-12-13-14",13));
testNodes.add(new TestNode(15,"2-5-9-10-11-12-13-14-15",14));
//建立多叉树进行遍历
MultiTree<TestNode> tree=new MultiTree<>();
List<TestNode> result=tree.getSortTreeList(testNodes);
for (TestNode node:result){
System.out.println(node);
}
}
}
使用递归进行遍历:《关于菜单树排序问题的解决方案(一)》
如有错误或者不足请指正,谢谢!