题目要求:

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

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 = item;
      }
  });
  return result;
}

妙阿!

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐