TypeScript实现数组和树的相互转换

// 定义数组项的数据类型，包含id、name、parentId基本属性
interface ArrayItem {
  id: number,
  name: string,
  parentId: number
}
// 定义树节点的数据类型，包含id、name、可能存在的子节点
interface TreeNode {
  id: number,
  name: string,
  children?: TreeNode[], // 叶子节点没有子节点
}
function array2Tree(arr: ArrayItem[]): TreeNode | null {
  // 用于id和TreeNode的映射，map<key, value>，可通过id快速查找到树节点，时间复杂度为O(1)
  const treeNode: Map<number, TreeNode> = new Map();
  let root = null; // 树根节点
  arr.forEach(item => {
    const {id, name, parentId} = item; // 解构赋值
    // 定义树节点tree node，并使用Map维持id与节点之间的关系
    const node: TreeNode = {id, name}
    treeNode.set(id, node);
​
    // 使用parentId找出parentNode
    const parentNode = treeNode.get(parentId);
    if (parentNode) {
      if (parentNode.children == null) {
        parentNode.children = [];
      }
      // 找到父节点后，将当前数组项转换的节点添加到子节点列表中
      parentNode.children.push(node)
    }
​
    // 在第一次循环中找到根节点，我们假定id=0的表示根节点
    if (parentId === 0) {
      root = node;
    }
  })
  return root;
}
const tree = array2Tree(arrs); // 需要定义arrs
console.log(tree);

// 定义数据结构
...
function tree2Array(root: TreeNode): ArrayItem[] {
  // 定义子节点和父节点的映射关系，map<子节点，父节点>
  const childToParent: Map<TreeNode, TreeNode> = new Map();

  const arr: ArrayItem[] = []; // 返回的数组

  // 广度优先遍历树，使用队列保存节点
  const queue: TreeNode[] = [];
  queue.unshift(root); // 入队，从头部入队
  while(queue.length) {
    const curNode = queue.pop(); // 出队，从尾部出队，保证先进先出
    if (curNode == null) break; // 循环结束，使用==包括了null和undefined的情况

    const {id, name, children = []} = curNode; // 结构赋值，叶子节点没有children属性，所以默认为空数组

    const parentNode = childToParent.get(curNode); // 获取当前节点的父节点
    const parentId = parentNode?.id || 0; // 获取父节点id，根节点设置为0
    const item = {id, name, parentId}; // 定义数组项
    arr.push(item); // 插入到数组中

    // 继续遍历子节点，并且子节点入队
    children.forEach(child => {
      // 如果有子节点，则把子节点和当前节点通过Map保持关系
      childToParent.set(child, curNode);

      // 入队
      queue.unshift(child);
    })
  }

  return arr;
}
const array = tree2Array(obj); // 这个obj对象就是数组转换为树打印出的结果
console.log(array)