非递归，谈谈如何将数组转换成树形结构？

前端

2024-01-02 07:21:47

前言

在编程工作中，我们经常会遇到需要处理树形结构数据的情况。树形结构是一种非线性的数据结构，可以用来表示具有层次关系的数据。常见的树形结构包括二叉树、红黑树和 B 树等。

在某些情况下，我们需要将数组转换成树形结构。例如，我们需要将一个包含员工信息的数组转换成一个组织架构树，或者将一个包含文件路径的数组转换成一个文件系统树。

非递归实现

将数组转换成树形结构可以通过递归和非递归两种方式来实现。递归是一种比较容易理解的方法，但是当数据量较大时，递归可能会导致栈溢出。因此，在实际应用中，我们通常使用非递归的方式来实现。

非递归实现的思路是使用一个队列来存储当前需要处理的节点。首先，我们将根节点放入队列中。然后，我们从队列中取出一个节点，并将其子节点放入队列中。重复这个过程，直到队列为空。

JavaScript 实现示例

function arrayToTree(arr) {
  // 根节点
  const root = {
    id: 0,
    children: [],
  };

  // 节点映射表
  const nodeMap = {};
  nodeMap[root.id] = root;

  // 队列
  const queue = [];
  queue.push(root);

  while (queue.length > 0) {
    // 从队列中取出一个节点
    const node = queue.shift();

    // 找到该节点的子节点
    const children = arr.filter(item => item.parentId === node.id);

    // 将子节点添加到该节点的 children 属性中
    node.children = children;

    // 将子节点放入队列中
    for (let i = 0; i < children.length; i++) {
      const child = children[i];
      nodeMap[child.id] = child;
      queue.push(child);
    }
  }

  return root;
}