如何将后端返回的js数组转为树形结构

function arrayToTree(array) {
  const map = new Map();
  const tree = [];

  // 将节点存储在 Map 中，以便可以按照 id 访问
  array.forEach(node => {
    map.set(node.id, { ...node, children: [] });
  });

  // 构建树状结构
  array.forEach(node => {
    const parent = map.get(node.parentId);
    if (parent) {
      parent.children.push(map.get(node.id));
    } else {
      tree.push(map.get(node.id));
    }
  });

  return tree;
}

// 示例数据
const nodes = [
  { id: 1, parentId: null },
  { id: 2, parentId: 1 },
  { id: 3, parentId: 1 },
  { id: 4, parentId: 2 },
  { id: 5, parentId: 2 },
  { id: 6, parentId: 3 },
  { id: 7, parentId: 3 },
];

const tree = arrayToTree(nodes);
console.log(tree);