递归 – 从路径字符串中获取树状结构

2023年5月25日 245次阅读

我被困2天,因为我不坚持指针和递归.我有一个类似路径的结构数组,让我们说：

s:=[]string {
  "a/b/c",
  "a/b/g",
  "a/d",
}

使用这样的数据结构：

 type Node struct {
   Name     string `json:"name"`
   Children []Node `json:"children"`
}

我想最终得到这样的东西：

{
 "name": "a",
 "children": [
     {
      "name": "b",
      "children": [
        {
         "name": "c",
         "children": []
        },
        {
         "name": "g",
         "children": []
        }
      ]
    },
    {
     "name": "d",
     "children": []
    }
  ]
}

我尝试使用递归来构建它,这种方法很好,但只有一个字符串(例如“a / b / c”),一旦我尝试实现应该添加缺少节点的东西(“g”in“ a / b / g“)我遇到了一棵树.

我有类似的东西：

func appendChild(root Node, children []string) Node {
   if len(children) == 1 {
      return Node{children[0], nil}
   } else {
      t := root
      t.Name=children[0]
      t.Children = append(t.Children, appendChild(root, children[1:]))
      return t
   }
}

有人能指出我有效的解决方案吗？

最佳答案
https://play.golang.org/p/9pER5cwChF

func AddToTree(root []Node, names []string) []Node {
    if len(names) > 0 {
        var i int
        for i = 0; i < len(root); i++ {
            if root[i].Name == names[0] { //already in tree
                break
            }
        }
        if i == len(root) {
            root = append(root, Node{Name: names[0]})
        }
        root[i].Children = AddToTree(root[i].Children, names[1:])
    }
    return root
}

示例输出(请注意我在children字段上使用omitempty,因为我不喜欢我的JSON中的空条目)：

[{
    "name": "a",
    "children": [{
        "name": "b",
        "children": [{
            "name": "c"
        }, {
            "name": "g"
        }]
    }, {
        "name": "d"
    }]
}]

与您的版本有显着差异：

>它在节点列表上运行,而不是单个节点的子节点.这很重要,因为您的版本假定所有树都具有相同的单根节点(a),而这可能不是这种情况.在你的版本中处理它的唯一方法是在根目录下有一个“假”节点.
>它不重用输入节点.这是您的代码的主要问题之一.如果len(孩子)> 1,更新输入节点的名称,附加到它的子节点,然后递归.这意味着切片的每个先前级别都成为孩子的一部分.您需要创建一个新节点.
>它实际上搜索树.您没有搜索树以查看正在插入的项目是否已存在,因此您复制节点(特别是节点b)

点赞