js实现数组转树形数据

题目要求：

将如下数组转化成树形结构：

const nodes = [
  { id: 3, name: '节点C', parentId: 1 },
  { id: 6, name: '节点F', parentId: 3 },
  { id: 0, name: 'root', parentId: null },
  { id: 1, name: '节点A', parentId: 0 },
  { id: 8, name: '节点H', parentId: 4 },
  { id: 4, name: '节点D', parentId: 1 },
  { id: 2, name: '节点B', parentId: 0 },
  { id: 5, name: '节点E', parentId: 2 },
  { id: 7, name: '节点G', parentId: 2 },
  { id: 9, name: '节点I', parentId: 5 }
];

树形结构为：

思路：

1、以id为key，构建一个对象obj，用来查找parent

2、遍历原数组，通过parentId从obj中找到父节点，然后把自己插入到父节点的children中

注：这里一定注意，对象是引用类型，所以map里边的对象改变，与之对应的原数组中的相同节点对象也会改变。

代码：

function convert(data) {
  let result;
  let map = {};
  data.forEach(item => {
      map[item.id] = item;
  });
  // console.log(map);
  data.forEach(item => {
      // item.pid 为null时 返回underfined
      let parent = map[item.parentId];
      if (parent) {
        (parent.children || (parent.children = [])).push(item);
      } else {
          // 这里push的item是pid为null的数据
          result.push(item);
      }
  });
  return result;
}

妙阿！