从路径字符串

时间:2017-05-05 13:09:38

标签: recursion go tree

我被困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 “在”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
   }
}

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

1 个答案:

答案 0 :(得分:1)

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
}

示例输出(请注意,我在子字段上使用了omitempty,因为我不喜欢我的JSON中的null条目:

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

与您的版本有显着差异:

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