从 GO 中的父/子对构建树结构

Question

我正在尝试从 Go 中的父 x 子数据集构建树结构。最终数据将来自 MySQL 表，但为了简化此示例，我已将数据库从等式中删除并创建了一个数组。“现实生活”解决方案的原理相同。

数据源如下所示：

--------------------------
    身份证 | 家长--------------------------
    1 | 0
    2 | 1
    3 | 1
    4 | 0
    5 | 4
    6 | 4
    7 | 0
    8 | 7
    9 | 2

The Parent colunm refer back to the ID column forming the parent x child relationship. ID's with parent = 0 are root elements.

这是我解决这个问题的尝试。它适用于根元素和第一级子元素，但它不会比树中的更深。我的理解是，我需要一个递归循环来实现我想要的，但我无法解决更深层次的问题。

https://play.golang.org/p/6m0YTqe529O

上述数据源的预期输出是一个分层树，如下所示：

（以 JSON 表示以方便可视化）。

[{

        "id": 1,

        "child": [{

                "id": 2,

                "child": [{

                    "id": 9

                }]

            },

            {

                "id": 3

            }

        ]

    },

    {

        "id": 4,

        "child": [{

            "id": 5

        }, {

            "id": 6

        }]

    },

    {

        "id": 7,

        "child": [{

            "id": 8

        }]

    }

]

我在 SO 上看到过类似的问题，但答案要么不高于第一级，要么根本不能完全解决问题。

Answer 1

首先，除非你真的必须这样做，否则你不应该使用指向切片的指针。为变量分配返回值更为典型，例如mySlice = sliceReturningFunction().我不确定这里的所有要求是什么，但一种解决方案可能是：构建父子关系图（map[int][]int）。将根级关系传递给递归构建类别的函数。这是一个示例递归函数。请注意，它返回一个新切片而不是改变一个指针。func buildCategories(ids []int, relations map[int][]int) []Category {    categories := make([]Category, len(ids))    for i, id := range ids {        c := Category{ID: id}        if childIDs, ok := relations[id]; ok {            c.Child = buildCategories(childIDs, relations)        }        categories[i] = c    }    return categories}我在 Playground 上添加了一个完整的示例。它没有经过测试，我相信有更好的解决方案，但它很简单，至少可以给你一些想法。如果您将拥有数千个节点并且经常访问这些节点，那么您将需要在 Go 代码之外进行优化。