如何在Node.js中基于MongoDB JSON数据构建无重复多树（森林）结构

当前代码的核心问题 #

你当前的实现会把每个节点都当作根节点生成独立树，导致子节点、孙子节点被重复生成多次，完全不符合森林结构的要求（每个节点只能属于一棵树，且仅出现一次）。

改进后的实现方案 #

下面的代码会自动识别根节点，且保证每个节点仅出现一次，高效构建森林结构：

const data = require("./nodes.json");

// 1. 构建节点ID到节点的映射表，实现O(1)快速查找
const nodeMap = new Map();
// 收集所有被其他节点引用的子节点ID
const referencedIds = new Set();

data.forEach(node => {
  const nodeId = node._id.$oid;
  nodeMap.set(nodeId, { ...node }); // 复制节点，避免修改原始数据
  // 将当前节点的所有子节点ID加入引用集合
  node.children.forEach(child => referencedIds.add(child.$oid));
});

// 2. 自动识别根节点：未被任何节点引用的节点就是根
const roots = data.filter(node => !referencedIds.has(node._id.$oid));

// 3. 递归构建单棵树，处理完的节点从映射表移除，避免重复/循环引用
function buildTree(nodeId) {
  const node = nodeMap.get(nodeId);
  if (!node) return null;

  // 移除已处理节点，防止重复构建或循环引用
  nodeMap.delete(nodeId);

  // 递归构建子节点列表，过滤无效节点
  const children = node.children
    .map(child => buildTree(child.$oid))
    .filter(child => child !== null);

  return {
    ...node,
    children
  };
}

// 构建完整森林（多树集合）
const forest = roots.map(root => buildTree(root._id.$oid));

// 改进的打印函数：支持遍历森林中所有树
function prettyPrintForest(forest, level = 0) {
  forest.forEach(root => {
    console.log(`${"  ".repeat(level)}(${root.kind}) ${root.name} ${root._id.$oid}`);
    prettyPrintNode(root, level + 1);
  });
}

function prettyPrintNode(node, level = 0) {
  if (node.children?.length) {
    node.children.forEach(child => {
      console.log(`${"  ".repeat(level)}(${child.kind}) ${child.name} ${child._id.$oid}`);
      prettyPrintNode(child, level + 1);
    });
  }
}

// 执行打印
prettyPrintForest(forest);

关键逻辑说明 #

1. 节点映射表：用Map存储节点，替代原代码中每次遍历数组的find操作，大幅提升查找效率。
2. 根节点识别：通过referencedIds集合记录所有被当作子节点的ID，未出现在集合中的节点就是树的根。
3. 无重复构建：从根节点出发递归构建子树，处理完的节点直接从映射表删除，确保每个节点仅被处理一次，同时避免循环引用问题。
4. 打印优化：新增prettyPrintForest函数，专门处理森林的多树打印需求，层级展示更清晰。

内容的提问来源于stack exchange，提问作者Yiffany