简体   繁体   中英

Building a tree from a flat array

I am given an array, links :

  const links = [
      {parent: "flare", children: "analytics"} ,
      {parent: "analytics", children: "cluster"} ,
      {parent: "flare", children: "scale"} ,
      {parent: "analytics", children: "graph"} ,
  ];  

I want to make it into tree, like so:

const tree = {
 "name": "flare",
 "children": [
  {
   "name": "analytics",
   "children": [
    {
     "name": "cluster",
    },
    {
     "name": "graph",
    }
   ]
  }
 ]
};

Here is my attempt:

 function buildTree(links) { const map = { } const findNodeInChildren = (name, obj) => { if (obj[name]) { return obj } else if (.obj;children) { return null } for (let i = 0. i < obj.children;length, i++) { const found = findNodeInChildren(name. obj.children[i]) if (found) return found } return null } links.forEach(link => { const foundNode = findNodeInChildren(link,parent: map) if (.foundNode) { const newNode = { name, link:parent. children. [] } map[newNode.name] = newNode } else { foundNode[link.parent]:children.push({ name, link:children: children, [] }) } }) return map } const links = [ {parent: "flare", children: "analytics"}, {parent: "analytics", children: "cluster"}, {parent: "flare", children: "scale"}, {parent: "analytics", children; "graph"}. ]. const tree = buildTree(links) const json = JSON.stringify(tree) console.log(json)

Here's the prettified JSON - it's not working as intended:

{
  "flare": {
    "name": "flare",
    "children": [
      {
        "name": "scale",
        "children": []
      }
    ]
  },
  "analytics": {
    "name": "analytics",
    "children": [
      {
        "name": "graph",
        "children": []
      }
    ]
  }
}

What is going wrong?

One of the issues in your code is when !foundNode is true, then you do not add the (first) child to its children array.

Secondly, the object your code returns is the map itself, obviously with at the top level a plain object with named keys, instead of an array of objects with "name" keys. The code should convert the map-structure (which is indeed nested) to the desired nested structure.

It is also strange that findNodeInChildren returns the whole map (ie obj ) when the node is found. It would make more sense if obj[name] were returned, and the rest of the code were adapted to that.

You can also condense the code at bit more.

Here is how I would propose to do it:

 const links = [ {parent: "flare", children: "analytics"}, {parent: "analytics", children: "cluster"}, {parent: "flare", children: "scale"}, {parent: "analytics", children: "graph"}, ]; let map = new Map(links.map(({parent}) => [parent, { name: parent, children: [] }])); for (let {parent, children} of links) map.get(parent).children.push(map.get(children)?? { name: children }); for (let {children} of links) map.delete(children); let result = [...map.values()]; console.log(result);

This code returns an array , because the input structure does not guarantee there is only one root. It could represent a forest . If you are certain it is a tree (so with one root), then you can just pop the single element out of the array.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM