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

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《在 GO 中从父/子对构建树结构》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

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

数据源如下所示：

---------------------------
    id      |   parent
---------------------------
    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
        }]
    }
]

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

解决方案

首先，除非确实必要，否则永远不应该使用指向切片的指针。更典型的是为变量分配一个返回值，例如myslice = slicereturningfunction()。

我不确定这里的所有要求是什么，但一种解决方案可能是：

1. 构建父子关系地图 (map[int][]int)。
2. 将根级关系传递给递归构建类别的函数。

这是一个递归函数的示例。请注意，它返回一个新切片而不是改变指针。

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
}

我添加了一个full example on the Playground。它没有经过测试，我确信有更好的解决方案，但它很简单，至少可以给你一些想法。如果您将拥有数千个节点并且经常访问这些节点，那么您将需要在 go 代码之外进行优化。

以上就是《在 GO 中从父/子对构建树结构》的详细内容，更多关于的资料请关注golang学习网公众号！