[2018][ruby][08] speedrun

sorbet
Alpha Chen 6 years ago
parent c807f716da
commit 6b086acd5b

@ -0,0 +1,34 @@
raw = ARGF.read.chomp.split(/\s+/).map(&:to_i)
class Node
def self.from(raw)
child_count = raw.shift
metadata_count = raw.shift
children = child_count.times.map { Node.from(raw) }
metadata = metadata_count.times.map { raw.shift }
Node.new(children, metadata)
end
def initialize(children, metadata)
@children, @metadata = children, metadata
end
def checksum
@metadata.sum + @children.map(&:checksum).sum
end
def value
if @children.empty?
@metadata.sum
else
@metadata.flat_map {|index|
@children.fetch(index-1, [])
}.map(&:value).sum
end
end
end
root = Node.from(raw)
p root.value
Loading…
Cancel
Save