关于菜单树排序问题的解决方案(二)

最近在做菜单管理的展示时,发现树形列表必须按照父在子前且子紧随父才能正确的展示,经过查阅资料后找到两种解决方案,一种是使用递归进行排序,这种方式请参考我的另一篇文章《关于菜单树排序问题的解决方案(一)》;另一种是使用多叉树进行遍历,本文讨论使用多叉树遍历的方案。我对这两种方案都简单封装了下,本文作为备忘。

使用多叉树对菜单树进行遍历

关于树节点接口和具体节点的实体请参考《关于菜单树排序问题的解决方案(一)》,我这里就不再重复了。

  1. 多叉树
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<>();
    }
}
  1. 多叉树方式的使用
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);
        }
    }
}

使用递归进行遍历:《关于菜单树排序问题的解决方案(一)》
如有错误或者不足请指正,谢谢!

推荐阅读更多精彩内容