从平面列表创建Java层次树集

如何解决从平面列表创建Java层次树集

这是我想出的解决方案

SortedSet<Node> nodeSet = new TreeSet<Node>(new Comparator<Node>() {
    public int compare(Node node1, Node node2) {

        if (node1.getParent() == null) {
            if (node2.getParent() == null) {
                return  node1.getId().compareto(node2.getId());
            }
            return -1;
        }

        if (node2.getParent() == null) return 1;

        int parentCompare = node1.getParent().getId()
                .compareto(node2.getParent().getId());

        if (parentCompare == 0)
            return node1.getId().compareto(node2.getId());

        return parentCompare;
    }
});

nodeSet.addAll(allData); // allData is the Node list


Map<Node, List<Node>> map = new HashMap<Node, List<Node>>();

for(Node node : nodeSet)
{
    if(map.get(node)==null)
    {
        map.put(node, new ArrayList<Node>());
    }
    map.get(node).add(node);
    Node parentNode = node.getParent();
    while(parentNode!=null)
    {
        map.get(parentNode).add(node);
        parentNode = parentNode.getParent();
    }
}

// At this point I can get an element from map and see all children in values.

解决方法

我有对象T的列表，它具有父属性，其中顶级对象的父属性为null。我想将所有对象放入TreeSet（或TreeMap）中。顶级对象将是所有没有父级的根对象（父级为null），并且它们的下级将是其子级。

像这样

              o
           /  |   \
          Ra  Rb   Rc          -- Level Root Objects
         / |   \    | \
        Ca1 Ca2 Cb1 Cc1 Cc2    -- Level of First Children
     /   \
   Ca11   Ca12..............   -- Level of Second Children

所以我可以得到Ra并找到它的子代（Ca1，Ca2，Ca11，Ca12…。）

更新：很抱歉，可能不清楚，节点指向父节点，如果parent为null，则它们是根节点。问题是父母需要了解孩子。但是这种关系是相反的。

class Node
{
  private Node parent;
  private String name;
}