记一道有意思的题目

昨天遇到了一道题目，一开始没解出来，后来再大佬的提示下做出来了。记录在这里。

题目

// 给定一个数组 arr，数组中的每个元素形式是
{
  id: 0,
  parent: 0,  // 指向数组中的另一个元素
  body: "xxx"
}

data:
[
  { id: 0, parent: 1, body: "wefw" },
  { id: 1, body: "sdwe" },
  { id: 2, parent: 1, body: "wefw" },
  { id: 3, body: "sdfew" },
  { id: 4, parent: 3, body: "we" },
  { id: 5, parent: 4, body: "2rd child" }
];

result:
[
  {
    id: 1,
    body: "sdwe",
    chidren: [{ id: 0, parent: 1, body: "wefw" }, { id: 2, parent: 1, body: "wefw" }]
  },
  {
    id: 3,
    body: "sdfew",
    children: [
      {
        id: 4,
        parent: "3",
        body: "we",
        children: [{ id: 5, parent: "4", body: "2rd child" }]
      }
    ]
  }
];
// 写一段代码，将这个数组变成一个森林，并且返回一个数组包括所有的根节点。
const result = arrayToTree(data);

思路

解这道题的时候有两个关键点：

1. 多个指针指向的是同一个对象；
2. 思路不应该从根开始找子，而是把子塞入父中。

解法

const arrayToTree = function(data) {
    let result = [];
    // 遍历数组得到一个idMap
    let idMap = {};
    data.map((item, index) => {
        idMap[item.id] = item;
    });
    // 遍历数组，有父节点的，就把他塞入父节点的children数组。
    data.map(item => {
        if (item.parent) {
            if (idMap[item.parent].children) {
                idMap[item.parent].children.push(item);
            } else {
                idMap[item.parent].children = [item];
            }
        }
    });
    // 此时的idMap就包含了所有的父子关系
    // 遍历找出所有的根节点，放入result即可得到结果
    for(key in idMap) {
        if (!idMap[key].parent) {
            result.push(idMap[key]);
        }
    }
    return result;
};