我正在编写一个Ruby脚本,想要使用一个n叉树数据结构。
是否有一个好的实现可作为源代码提供?谢谢。
在继承Otto的答案上,让哈希表自动创建数组的一种简单方法是使用默认值块和 Hash::new
方法,如下所示:
node_children = Hash.new { |_node_children, node_key| _node_children[node_key] = [] }
但是实际上,代码取决于您想要对数组执行什么操作。您可以使用哈希和数组设置它们,也可以创建一些类:
class Node
attr_accessor :value, :children
def initialize(value, children=[])
@value = value
@children = children
end
def to_s(indent=0)
value_s = @value.to_s
sub_indent = indent + value_s.length
value_s + @children.map { |child| " - " + child.to_s(sub_indent + 3) }.join("\n" + ' ' * sub_indent)
end
end
ROOT = Node.new('root', %w{ farleft left center right farright }.map { |str| Node.new(str) } )
puts "Original Tree"
puts ROOT
puts
ROOT.children.each do |node|
node.children = %w{ one two three four }.map { |str| Node.new(node.value + ':' + str) }
end
puts "New Tree"
puts ROOT
puts
例如,此代码会产生:
Original Tree
root - farleft
- left
- center
- right
- farright
New Tree
root - farleft - farleft:one
- farleft:two
- farleft:three
- farleft:four
- left - left:one
- left:two
- left:three
- left:four
- center - center:one
- center:two
- center:three
- center:four
- right - right:one
- right:two
- right:three
- right:four
- farright - farright:one
- farright:two
- farright:three
- farright:four