Ruby实现Neo4j查询结果转D3树形嵌套数组方法求助

核心思路 #

不用递归遍历查找父节点位置，而是先通过哈希表建立节点索引，实现O(1)快速定位父节点，再批量绑定父子关系，最后筛选出根节点即可得到D3需要的嵌套结构。

分步实现 #

1. 构建节点索引哈希

先把所有节点从Neo4j返回的二维数组中提取出来，为每个节点添加空的children数组，并用节点id作为键存入哈希表，方便后续快速查找：

# 假设neo4j_results是你拿到的二维数组结果
nodes_hash = {}

# 提取所有节点并初始化children
neo4j_results.flatten.each do |raw_node|
  nodes_hash[raw_node[:id]] = raw_node.merge(children: [])
end

2. 绑定父子关系

遍历每一组父-子关系数组，将子节点批量添加到对应父节点的children数组中：

neo4j_results.each do |relation_group|
  # 取当前组的父节点（数组第一个元素）
  parent_node = nodes_hash[relation_group[0][:id]]
  # 遍历当前组的所有子节点（数组从第二个元素开始）
  relation_group[1..-1].each do |raw_child|
    child_node = nodes_hash[raw_child[:id]]
    parent_node[:children] << child_node
  end
end

3. 筛选根节点

根节点是没有被任何其他节点作为子节点的节点，通过集合快速筛选：

# 收集所有子节点的id
all_child_ids = neo4j_results.flat_map { |group| group[1..-1] }.map { |n| n[:id] }.to_set

# 找出不在子节点集合里的节点，就是根节点（单根树取first，多根树直接用数组）
root = nodes_hash.values.reject { |node| all_child_ids.include?(node[:id]) }.first

完整示例 #

假设你的Neo4j返回数据如下：

neo4j_results = [
  [{ id: 1, name: "Root" }, { id: 2, name: "Child 1" }],
  [{ id: 1, name: "Root" }, { id: 3, name: "Child 2" }],
  [{ id: 2, name: "Child 1" }, { id: 4, name: "Grandchild 1" }]
]

运行上述代码后，root的结构就是D3所需的格式：

{
  id: 1,
  name: "Root",
  children: [
    {
      id: 2,
      name: "Child 1",
      children: [
        { id: 4, name: "Grandchild 1", children: [] }
      ]
    },
    { id: 3, name: "Child 2", children: [] }
  ]
}

优势说明 #

• 效率高：哈希表查找是O(1)时间复杂度，比递归遍历树结构的O(n)查找快得多，适合大规模数据
• 逻辑清晰：分三步处理，避免递归嵌套的复杂逻辑，易维护和调试

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