【问题标题】:How to efficiently build a tree from a flat structure?如何有效地从平面结构中构建树?
【发布时间】:2010-10-01 10:51:57
【问题描述】:

我有一堆扁平结构的对象。这些对象具有IDParentID 属性,因此它们可以排列在树中。它们没有特定的顺序。 每个ParentID 属性不一定与结构中的ID 匹配。因此它们可能是从这些物体中长出的几棵树。

您将如何处理这些对象以创建结果树?

我离解决方案不远,但我确信它远非最佳......

我需要创建这些树,然后以正确的顺序将数据插入数据库。

没有循环引用。当 ParentID == null 或在其他对象中找不到 ParentID 时,Node 是 RootNode

【问题讨论】:

  • “创造”是什么意思?在 UI 中渲染?以分层方式存储在 XML 或数据库中?
  • 如何定义没有父节点的节点(即根节点)。 ParentID 为空?父母 ID = 0?我假设没有正确的循环引用?
  • 我觉得这个问题很酷。

标签: algorithm tree language-agnostic hierarchical-data


【解决方案1】:

将对象的 ID 存储在映射到特定对象的哈希表中。枚举所有对象并找到它们的父对象(如果存在)并相应地更新其父指针。

class MyObject
{ // The actual object
    public int ParentID { get; set; }
    public int ID { get; set; }
}

class Node
{
    public List<Node> Children = new List<Node>();
    public Node Parent { get; set; }
    public MyObject AssociatedObject { get; set; }
}

IEnumerable<Node> BuildTreeAndGetRoots(List<MyObject> actualObjects)
{
    Dictionary<int, Node> lookup = new Dictionary<int, Node>();
    actualObjects.ForEach(x => lookup.Add(x.ID, new Node { AssociatedObject = x }));
    foreach (var item in lookup.Values) {
        Node proposedParent;
        if (lookup.TryGetValue(item.AssociatedObject.ParentID, out proposedParent)) {
            item.Parent = proposedParent;
            proposedParent.Children.Add(item);
        }
    }
    return lookup.Values.Where(x => x.Parent == null);
}

【讨论】:

  • 那是什么语言? (我认为它是 C#)
  • 这个算法是(非正式表示法)O(3N),其中 O(1N) 解决方案很容易通过为非“遍历”父级实例化部分节点或保持辅助外观来实现 -为非实例化父母的孩子准备桌子。可能对大多数实际用途无关紧要,但在大型数据集上可能很重要。
  • @AndrewHanlon 也许你应该发布 0(1N) 的 sol
  • @Ced Martin Schmidt 下面的回答与我的实现方式非常接近。可以看出,它使用单个循环,其余都是哈希表操作。
  • O(3N) 只是 O(N) ;)
【解决方案2】:

这是一个简单的 JavaScript 算法,用于将平面表解析为 N 次运行的父/子树结构:

var table = [
    {parent_id: 0, id: 1, children: []},
    {parent_id: 0, id: 2, children: []},
    {parent_id: 0, id: 3, children: []},
    {parent_id: 1, id: 4, children: []},
    {parent_id: 1, id: 5, children: []},
    {parent_id: 1, id: 6, children: []},
    {parent_id: 2, id: 7, children: []},
    {parent_id: 7, id: 8, children: []},
    {parent_id: 8, id: 9, children: []},
    {parent_id: 3, id: 10, children: []}
];

var root = {id:0, parent_id: null, children: []};
var node_list = { 0 : root};

for (var i = 0; i < table.length; i++) {
    node_list[table[i].id] = table[i];
    node_list[table[i].parent_id].children.push(node_list[table[i].id]);
}

console.log(root);

【讨论】:

  • 试图将此方法转换为 C#。
  • 意识到如果 id 从 1001 之类的大的东西开始,那么我们会得到 index out of bound 异常...
  • 提示:使用console.log(JSON.stringify(root, null, 2)); 漂亮地打印输出。
  • 如果节点没有按父ID排序,这将失败
【解决方案3】:

根据 Mehrdad Afshari 的回答和 Andrew Hanlon 对加速的评论,这是我的看法。

与原始任务的重要区别:根节点的 ID==parentID。

class MyObject
{   // The actual object
    public int ParentID { get; set; }
    public int ID { get; set; }
}

class Node
{
    public List<Node> Children = new List<Node>();
    public Node Parent { get; set; }
    public MyObject Source { get; set; }
}

List<Node> BuildTreeAndGetRoots(List<MyObject> actualObjects)
{
    var lookup = new Dictionary<int, Node>();
    var rootNodes = new List<Node>();

    foreach (var item in actualObjects)
    {
        // add us to lookup
        Node ourNode;
        if (lookup.TryGetValue(item.ID, out ourNode))
        {   // was already found as a parent - register the actual object
            ourNode.Source = item;
        }
        else
        {
            ourNode = new Node() { Source = item };
            lookup.Add(item.ID, ourNode);
        }

        // hook into parent
        if (item.ParentID == item.ID)
        {   // is a root node
            rootNodes.Add(ourNode);
        }
        else
        {   // is a child row - so we have a parent
            Node parentNode;
            if (!lookup.TryGetValue(item.ParentID, out parentNode))
            {   // unknown parent, construct preliminary parent
                parentNode = new Node();
                lookup.Add(item.ParentID, parentNode);
            }
            parentNode.Children.Add(ourNode);
            ourNode.Parent = parentNode;
        }
    }

    return rootNodes;
}

【讨论】:

  • 很好,这基本上就是我所暗示的方法。但是,我将只使用一个伪根节点(ID = 0 和 null Parent)并删除自引用要求。
  • 此示例中唯一缺少的是将 Parent 字段分配给每个子节点。为此,我们只需在将子项添加到其父集合后设置父字段。像这样: parentNode.Children.Add(ourNode); ourNode.Parent = parentNode;
  • @plauriola 真的,谢谢,我添加了这个。另一种方法是仅删除 Parent 属性,核心算法不需要。
  • 由于我找不到一个实现 O(n) 解决方案的 npm 模块,我创建了以下一个(单元测试,100% 代码覆盖率,大小只有 0.5 kb 并且包括打字。也许它帮助某人:npmjs.com/package/performant-array-to-tree
【解决方案4】:

Python 解决方案

    def subtree(node, relationships):
        return {
            v: subtree(v, relationships) 
            for v in [x[0] for x in relationships if x[1] == node]
        }

例如:

    # (child, parent) pairs where -1 means no parent    
    flat_tree = [
         (1, -1),
         (4, 1),
         (10, 4),
         (11, 4),
         (16, 11),
         (17, 11),
         (24, 17),
         (25, 17),
         (5, 1),
         (8, 5),
         (9, 5),
         (7, 9),
         (12, 9),
         (22, 12),
         (23, 12),
         (2, 23),
         (26, 23),
         (27, 23),
         (20, 9),
         (21, 9)
        ]
    
    subtree(-1, flat_tree)

生产:

    {
        "1": {
            "4": {
                "10": {}, 
                "11": {
                    "16": {}, 
                    "17": {
                        "24": {}, 
                        "25": {}
                    }
                }
            }, 
            "5": {
                "8": {}, 
                "9": {
                    "20": {}, 
                    "12": {
                        "22": {}, 
                        "23": {
                            "2": {}, 
                            "27": {}, 
                            "26": {}
                        }
                    }, 
                    "21": {}, 
                    "7": {}
                }
            }
        }
    }

【讨论】:

  • 嗨。如何在输出中添加另一个属性? IE。姓名,父母ID
  • 迄今为止最优雅的!
  • @simpleguy:列表推导可以展开以防您需要更多控制,例如:def recurse(id, pages): for row in rows: if row['id'] == id: print(f'''{row['id']}:{row['parent_id']} {row['path']} {row['title']}''') recurse(row['id'], rows)
【解决方案5】:

返回一个根或根数组的 JS 版本,每个根都有一个包含相关子项的 Children 数组属性。不依赖于有序输入,相当紧凑,并且不使用递归。享受!

// creates a tree from a flat set of hierarchically related data
var MiracleGrow = function(treeData, key, parentKey)
{
    var keys = [];
    treeData.map(function(x){
        x.Children = [];
        keys.push(x[key]);
    });
    var roots = treeData.filter(function(x){return keys.indexOf(x[parentKey])==-1});
    var nodes = [];
    roots.map(function(x){nodes.push(x)});
    while(nodes.length > 0)
    {

        var node = nodes.pop();
        var children =  treeData.filter(function(x){return x[parentKey] == node[key]});
        children.map(function(x){
            node.Children.push(x);
            nodes.push(x)
        });
    }
    if (roots.length==1) return roots[0];
    return roots;
}


// demo/test data
var treeData = [

    {id:9, name:'Led Zep', parent:null},
    {id:10, name:'Jimmy', parent:9},
    {id:11, name:'Robert', parent:9},
    {id:12, name:'John', parent:9},

    {id:8, name:'Elec Gtr Strings', parent:5},
    {id:1, name:'Rush', parent:null},
    {id:2, name:'Alex', parent:1},
    {id:3, name:'Geddy', parent:1},
    {id:4, name:'Neil', parent:1},
    {id:5, name:'Gibson Les Paul', parent:2},
    {id:6, name:'Pearl Kit', parent:4},
    {id:7, name:'Rickenbacker', parent:3},

    {id:100, name:'Santa', parent:99},
    {id:101, name:'Elf', parent:100},

];
var root = MiracleGrow(treeData, "id", "parent")
console.log(root)

【讨论】:

  • 这个问题已有 7 年历史,已经有一个投票和接受的答案。如果您认为自己有更好的解决方案,最好在您的代码中添加一些解释。
  • 这种方法适用于这种无序类型的数据。
【解决方案6】:

对于任何对 Eugene 解决方案的 C# 版本感兴趣的人,请注意 node_list 作为地图访问,因此请改用字典。

请记住,此解决方案仅适用于 tableparent_id 排序的情况。

var table = new[]
{
    new Node { parent_id = 0, id = 1 },
    new Node { parent_id = 0, id = 2 },
    new Node { parent_id = 0, id = 3 },
    new Node { parent_id = 1, id = 4 },
    new Node { parent_id = 1, id = 5 },
    new Node { parent_id = 1, id = 6 },
    new Node { parent_id = 2, id = 7 },
    new Node { parent_id = 7, id = 8 },
    new Node { parent_id = 8, id = 9 },
    new Node { parent_id = 3, id = 10 },
};

var root = new Node { id = 0 };
var node_list = new Dictionary<int, Node>{
    { 0, root }
};

foreach (var item in table)
{
    node_list.Add(item.id, item);
    node_list[item.parent_id].children.Add(node_list[item.id]);
}

节点定义如下。

class Node
{
    public int id { get; set; }
    public int parent_id { get; set; }
    public List<Node> children = new List<Node>();
}

【讨论】:

  • 它太旧了,但是列表项 8 new Node { parent_id = 7, id = 9 }, 阻止 node_list.Add(item.id, item); 完成,因为 Key 不能重复;这是一个错字;因此,输入 id = 8 而不是 id = 9
  • 已修复。谢谢@MarceloScofano!
  • 看起来随机节点顺序会失败。 (例如,当根节点结束时)
【解决方案7】:

在这里找到了一个很棒的 JavaScript 版本:http://oskarhane.com/create-a-nested-array-recursively-in-javascript/

假设你有一个这样的数组:

const models = [
    {id: 1, title: 'hello', parent: 0},
    {id: 2, title: 'hello', parent: 0},
    {id: 3, title: 'hello', parent: 1},
    {id: 4, title: 'hello', parent: 3},
    {id: 5, title: 'hello', parent: 4},
    {id: 6, title: 'hello', parent: 4},
    {id: 7, title: 'hello', parent: 3},
    {id: 8, title: 'hello', parent: 2}
];

你想让对象像这样嵌套:

const nestedStructure = [
    {
        id: 1, title: 'hello', parent: 0, children: [
            {
                id: 3, title: 'hello', parent: 1, children: [
                    {
                        id: 4, title: 'hello', parent: 3, children: [
                            {id: 5, title: 'hello', parent: 4},
                            {id: 6, title: 'hello', parent: 4}
                        ]
                    },
                    {id: 7, title: 'hello', parent: 3}
                ]
            }
        ]
    },
    {
        id: 2, title: 'hello', parent: 0, children: [
            {id: 8, title: 'hello', parent: 2}
        ]
    }
];

这是实现它的递归函数。

function getNestedChildren(models, parentId) {
    const nestedTreeStructure = [];
    const length = models.length;

    for (let i = 0; i < length; i++) { // for-loop for perf reasons, huge difference in ie11
        const model = models[i];

        if (model.parent == parentId) {
            const children = getNestedChildren(models, model.id);

            if (children.length > 0) {
                model.children = children;
            }

            nestedTreeStructure.push(model);
        }
    }

    return nestedTreeStructure;
}

用法:

const models = [
    {id: 1, title: 'hello', parent: 0},
    {id: 2, title: 'hello', parent: 0},
    {id: 3, title: 'hello', parent: 1},
    {id: 4, title: 'hello', parent: 3},
    {id: 5, title: 'hello', parent: 4},
    {id: 6, title: 'hello', parent: 4},
    {id: 7, title: 'hello', parent: 3},
    {id: 8, title: 'hello', parent: 2}
];
const nestedStructure = getNestedChildren(models, 0);

【讨论】:

  • 对于每个 parentId 它都在循环整个模型 - 这不是 O(N^2) 吗?
【解决方案8】:

这是 Mehrdad Afshari 的答案的 java 解决方案。

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;

public class Tree {

    Iterator<Node> buildTreeAndGetRoots(List<MyObject> actualObjects) {
        Map<Integer, Node> lookup = new HashMap<>();
        actualObjects.forEach(x -> lookup.put(x.id, new Node(x)));
        //foreach (var item in lookup.Values)
        lookup.values().forEach(item ->
                {
                    Node proposedParent;
                    if (lookup.containsKey(item.associatedObject.parentId)) {
                        proposedParent = lookup.get(item.associatedObject.parentId);
                        item.parent = proposedParent;
                        proposedParent.children.add(item);
                    }
                }
        );
        //return lookup.values.Where(x =>x.Parent ==null);
        return lookup.values().stream().filter(x -> x.parent == null).iterator();
    }

}

class MyObject { // The actual object
    public int parentId;
    public int id;
}

class Node {
    public List<Node> children = new ArrayList<Node>();
    public Node parent;
    public MyObject associatedObject;

    public Node(MyObject associatedObject) {
        this.associatedObject = associatedObject;
    }
}

【讨论】:

  • 你应该解释一下你在代码背后的想法。
  • 这只是早期答案的Java翻译
【解决方案9】:

我根据@Mehrdad Afshari 的回答用 C# 写了一个通用的解决方案:

void Example(List<MyObject> actualObjects)
{
  List<TreeNode<MyObject>> treeRoots = actualObjects.BuildTree(obj => obj.ID, obj => obj.ParentID, -1);
}

public class TreeNode<T>
{
  public TreeNode(T value)
  {
    Value = value;
    Children = new List<TreeNode<T>>();
  }

  public T Value { get; private set; }
  public List<TreeNode<T>> Children { get; private set; }
}

public static class TreeExtensions
{
  public static List<TreeNode<TValue>> BuildTree<TKey, TValue>(this IEnumerable<TValue> objects, Func<TValue, TKey> keySelector, Func<TValue, TKey> parentKeySelector, TKey defaultKey = default(TKey))
  {
    var roots = new List<TreeNode<TValue>>();
    var allNodes = objects.Select(overrideValue => new TreeNode<TValue>(overrideValue)).ToArray();
    var nodesByRowId = allNodes.ToDictionary(node => keySelector(node.Value));

    foreach (var currentNode in allNodes)
    {
      TKey parentKey = parentKeySelector(currentNode.Value);
      if (Equals(parentKey, defaultKey))
      {
        roots.Add(currentNode);
      }
      else
      {
        nodesByRowId[parentKey].Children.Add(currentNode);
      }
    }

    return roots;
  }
}

【讨论】:

  • 投反对票,请发表评论。我很高兴知道我做错了什么。
【解决方案10】:

在我看来这个问题含糊不清,我可能会创建一个从 ID 到实际对象的映射。在伪java中(我没有检查它是否工作/编译),它可能是这样的:

Map<ID, FlatObject> flatObjectMap = new HashMap<ID, FlatObject>();

for (FlatObject object: flatStructure) {
    flatObjectMap.put(object.ID, object);
}

并查找每个父母:

private FlatObject getParent(FlatObject object) {
    getRealObject(object.ParentID);
}

private FlatObject getRealObject(ID objectID) {
    flatObjectMap.get(objectID);
}

通过重用 getRealObject(ID) 并从对象映射到对象集合(或它们的 ID),您也可以获得一个父->子映射。

【讨论】:

    【解决方案11】:

    假设 Dictionary 类似于 TreeMap,我可以在 4 行代码和 O(n log n) 时间内完成此操作。

    dict := Dictionary new.
    ary do: [:each | dict at: each id put: each].
    ary do: [:each | (dict at: each parent) addChild: each].
    root := dict at: nil.
    

    编辑: 好的,现在我读到一些 parentID 是假的,所以忘记上面的,然后这样做:

    dict := Dictionary new.
    dict at: nil put: OrderedCollection new.
    ary do: [:each | dict at: each id put: each].
    ary do: [:each | 
        (dict at: each parent ifAbsent: [dict at: nil]) 
              add: each].
    roots := dict at: nil.
    

    【讨论】:

      【解决方案12】:

      大多数答案都假设您希望在数据库之外执行此操作。如果您的树本质上是相对静态的,并且您只需要以某种方式将树映射到数据库中,您可能需要考虑在数据库端使用嵌套集表示。查看 Joe Celko 的书籍(或 here Celko 的概述)。

      如果仍然绑定到 Oracle dbs,请查看他们的 CONNECT BY 以获得直接的 SQL 方法。

      无论使用哪种方法,您都可以在将数据加载到数据库之前完全跳过树的映射。只是想我会提供这个作为替代方案,它可能完全不适合您的特定需求。原始问题的整个“正确顺序”部分在某种程度上暗示您出于某种原因需要数据库中的“正确”顺序?这也可能促使我去处理那里的树木。

      【讨论】:

        【解决方案13】:

        这与提问者所寻找的并不完全相同,但我很难理解这里提供的措辞含糊的答案,我仍然认为这个答案符合标题。

        我的答案是将平面结构映射到直接在对象上的树,其中您所拥有的只是每个对象上的 ParentIDParentIDnull0 如果它是根。与提问者相反,我假设所有有效的ParentID 指向列表中的其他内容:

        var rootNodes = new List<DTIntranetMenuItem>();
        var dictIntranetMenuItems = new Dictionary<long, DTIntranetMenuItem>();
        
        //Convert the flat database items to the DTO's,
        //that has a list of children instead of a ParentID.
        foreach (var efIntranetMenuItem in flatIntranetMenuItems) //List<tblIntranetMenuItem>
        {
            //Automapper (nuget)
            DTIntranetMenuItem intranetMenuItem =
                                           Mapper.Map<DTIntranetMenuItem>(efIntranetMenuItem);
            intranetMenuItem.Children = new List<DTIntranetMenuItem>();
            dictIntranetMenuItems.Add(efIntranetMenuItem.ID, intranetMenuItem);
        }
        
        foreach (var efIntranetMenuItem in flatIntranetMenuItems)
        {
            //Getting the equivalent object of the converted ones
            DTIntranetMenuItem intranetMenuItem = dictIntranetMenuItems[efIntranetMenuItem.ID];
        
            if (efIntranetMenuItem.ParentID == null || efIntranetMenuItem.ParentID <= 0)
            {
                rootNodes.Add(intranetMenuItem);
            }
            else
            {
                var parent = dictIntranetMenuItems[efIntranetMenuItem.ParentID.Value];
                parent.Children.Add(intranetMenuItem);
                //intranetMenuItem.Parent = parent;
            }
        }
        return rootNodes;
        

        【讨论】:

          【解决方案14】:

          这是一个 ruby​​ 实现:

          它将按属性名称或方法调用的结果进行分类。

          CatalogGenerator = ->(depth) do
            if depth != 0
              ->(hash, key) do
                hash[key] = Hash.new(&CatalogGenerator[depth - 1])
              end
            else
              ->(hash, key) do
                hash[key] = []
              end
            end
          end
          
          def catalog(collection, root_name: :root, by:)
            method_names = [*by]
            log = Hash.new(&CatalogGenerator[method_names.length])
            tree = collection.each_with_object(log) do |item, catalog|
              path = method_names.map { |method_name| item.public_send(method_name)}.unshift(root_name.to_sym)
            catalog.dig(*path) << item
            end
            tree.with_indifferent_access
          end
          
           students = [#<Student:0x007f891d0b4818 id: 33999, status: "on_hold", tenant_id: 95>,
           #<Student:0x007f891d0b4570 id: 7635, status: "on_hold", tenant_id: 6>,
           #<Student:0x007f891d0b42c8 id: 37220, status: "on_hold", tenant_id: 6>,
           #<Student:0x007f891d0b4020 id: 3444, status: "ready_for_match", tenant_id: 15>,
           #<Student:0x007f8931d5ab58 id: 25166, status: "in_partnership", tenant_id: 10>]
          
          catalog students, by: [:tenant_id, :status]
          
          # this would out put the following
          {"root"=>
            {95=>
              {"on_hold"=>
                [#<Student:0x007f891d0b4818
                  id: 33999,
                  status: "on_hold",
                  tenant_id: 95>]},
             6=>
              {"on_hold"=>
                [#<Student:0x007f891d0b4570 id: 7635, status: "on_hold", tenant_id: 6>,
                 #<Student:0x007f891d0b42c8
                  id: 37220,
                  status: "on_hold",
                  tenant_id: 6>]},
             15=>
              {"ready_for_match"=>
                [#<Student:0x007f891d0b4020
                  id: 3444,
                  status: "ready_for_match",
                  tenant_id: 15>]},
             10=>
              {"in_partnership"=>
                [#<Student:0x007f8931d5ab58
                  id: 25166,
                  status: "in_partnership",
                  tenant_id: 10>]}}}
          

          【讨论】:

            【解决方案15】:

            接受的答案对我来说太复杂了,所以我添加了它的 Ruby 和 NodeJS 版本

            假设平面节点列表具有以下结构:

            nodes = [
              { id: 7, parent_id: 1 },
              ...
            ] # ruby
            
            nodes = [
              { id: 7, parentId: 1 },
              ...
            ] # nodeJS
            

            将上面的平面列表结构变成树的函数如下所示

            对于红宝石:

            def to_tree(nodes)
            
              nodes.each do |node|
            
                parent = nodes.find { |another| another[:id] == node[:parent_id] }
                next unless parent
            
                node[:parent] = parent
                parent[:children] ||= []
                parent[:children] << node
            
              end
            
              nodes.select { |node| node[:parent].nil? }
            
            end
            

            对于 NodeJS:

            const toTree = (nodes) => {
            
              nodes.forEach((node) => {
            
                const parent = nodes.find((another) => another.id == node.parentId)
                if(parent == null) return;
            
                node.parent = parent;
                parent.children = parent.children || [];
                parent.children = parent.children.concat(node);
            
              });
            
              return nodes.filter((node) => node.parent == null)
            
            };
            

            【讨论】:

            • 我相信null的检查需要undefined
            • @Ullauri null == undefined =&gt; true 在 NodeJS 中
            【解决方案16】:

            做到这一点的一种优雅方式是将列表中的项目表示为字符串,其中包含一个点分隔的父列表,最后是一个值:

            server.port=90
            server.hostname=localhost
            client.serverport=90
            client.database.port=1234
            client.database.host=localhost
            

            当组装一棵树时,你最终会得到类似的东西:

            server:
              port: 90
              hostname: localhost
            client:
              serverport=1234
              database:
                port: 1234
                host: localhost
            

            我有一个configuration library,它从命令行参数(列表)实现了这个覆盖配置(树)。 将列表中的单个项目添加到树 is here 的算法。

            【讨论】:

              【解决方案17】:

              您是否只使用这些属性?如果没有,最好创建一个子节点数组,您可以在其中循环遍历所有这些对象以构建此类属性。从那里,选择有子节点但没有父节点的节点,然后从上到下迭代构建树。

              【讨论】:

                【解决方案18】:

                java版本

                // node
                @Data
                public class Node {
                    private Long id;
                    private Long parentId;
                    private String name;
                    private List<Node> children = new ArrayList<>();
                }
                
                // flat list to tree
                List<Node> nodes = new ArrayList();// load nodes from db or network
                Map<Long, Node> nodeMap = new HashMap();
                nodes.forEach(node -> {
                  if (!nodeMap.containsKey(node.getId)) nodeMap.put(node.getId, node);
                  if (nodeMap.containsKey(node.getParentId)) {
                    Node parent = nodeMap.get(node.getParentId);
                    node.setParentId(parent.getId());
                    parent.getChildren().add(node);
                  }
                });
                
                // tree node
                List<Node> treeNode = nodeMap .values().stream().filter(n -> n.getParentId() == null).collect(Collectors.toList());
                

                【讨论】:

                  【解决方案19】:

                  请参阅下面带有简单测试程序的完整 Java 8+ 解决方案。

                  这是一种对@“Vimal Bhatt”版本进行修改的解决方案,该版本接受多个根

                  package tree;
                  
                  import java.util.*;
                  import java.util.function.Consumer;
                  import java.util.stream.Stream;
                  
                  public class Tree {
                      
                      private <T> void swap(T[] input, int a, int b) {
                          T tmp = input[a];
                          input[a] = input[b];
                          input[b] = tmp;
                      }
                      
                      public static void main(String[] args) {
                          Random r = new Random(8);
                  
                          MyObject[] myObjects = new MyObject[]{
                                  new MyObject(6, 3),
                                  new MyObject(7, 5),
                                  new MyObject(8, 0),
                                  new MyObject(1, 0),
                                  new MyObject(15, 12),
                                  new MyObject(12, 0),
                                  new MyObject(3, 5),
                                  new MyObject(4, 3),
                                  new MyObject(5, 2),
                                  new MyObject(2, 1),
                                  new MyObject(21, 8),
                                  new MyObject(9, 1)
                          };
                          
                          Tree t = new Tree();
                          // cinco trocas arbitrarias de posição
                          for (int i = 0; i < 5; i++) {
                              int a = r.nextInt(7) + 1;
                              int b = r.nextInt(7) + 1;
                              t.swap(myObjects, a, b);
                          }
                          System.out.println("The list have " + myObjects.length + " objects");
                          for (MyObject o: myObjects) {
                              System.out.print(" " + o);
                          }
                          Iterator<Node> iterator = t.buildTreeAndGetRoots(Arrays.asList(myObjects));
                          int counter = 0;
                          System.out.println("");
                          while (iterator.hasNext()) {
                              Node obj = iterator.next();
                              System.out.println(++counter + "\t" + obj.associatedObject.id + "\t-> " + obj.associatedObject.parentId);
                          }
                      }
                  
                      Iterator<Node> buildTreeAndGetRoots(List<MyObject> actualObjects) {
                          Node root = null;
                          Map<Integer, Node> lookup = new HashMap<>();
                          actualObjects.forEach(x -> lookup.put(x.id, new Node(x)));
                  
                          Stream<Node> roots = actualObjects.stream()
                                  .filter(x -> x.parentId == 0)
                                  .map(x -> new Node(x));
                          Consumer<Node> nodeConsumer = item -> {
                              Node proposedParent;
                              if (lookup.containsKey(item.associatedObject.parentId)) {
                                  proposedParent = lookup.get(item.associatedObject.parentId);
                                  item.parent = proposedParent;
                                  proposedParent.children.add(item);
                              }
                          };
                          lookup.values().forEach(nodeConsumer);
                          Stream<Node> s2 = lookup.values().stream().filter(e -> e.associatedObject.parentId != 0);
                          return Stream.concat(roots, s2).iterator();
                      }
                  }
                  
                  class MyObject { // The actual object
                      public int parentId;
                      public int id;
                  
                      MyObject(int id, int parent) {
                          this.parentId = parent;
                          this.id = id;
                      }
                  
                      @Override
                      public String toString() {
                          return "{ " +
                                  "parent: " + parentId +
                                  ", id: " + id +
                                  " }" ;
                      }
                  }
                  
                  class Node {
                      public List<Node> children = new ArrayList<Node>();
                      public Node parent;
                      public MyObject associatedObject;
                  
                      public Node(MyObject associatedObject) {
                          this.associatedObject = associatedObject;
                      }
                  }
                  
                  

                  【讨论】:

                    【解决方案20】:

                    此解决方案与所选答案相同,但在 JavaScript 中:

                    /**
                     * @param {{id: any, parentId: any}[]} nodes
                     */
                    function arrayToTree(nodes) {
                      const map = new Map(nodes.map((x) => [x.id, { key: x.id, children: [] }]));
                      for (const n of nodes) {
                        if (n.parentId) {
                          map.get(n.parentId)?.children?.push(map.get(n.id));
                        }
                      }
                      return nodes.filter((x) => x.parentId === null).map((x) => map.get(x.id));
                    }
                    

                    【讨论】:

                      猜你喜欢
                      • 1970-01-01
                      • 2022-01-07
                      • 1970-01-01
                      • 1970-01-01
                      • 2017-07-19
                      • 2021-04-19
                      • 2018-12-01
                      • 1970-01-01
                      • 1970-01-01
                      相关资源
                      最近更新 更多