08/2 - Ruby

Leo Silva Souza · Leo Silva Souza · commit 355da959efd1 · 2018-12-08T18:15:16.000-02:00
diff --git a/2018/08_2/ruby/leovano.rb b/2018/08_2/ruby/leovano.rb
@@ -0,0 +1,28 @@
+# frozen_string_literal: true
+
+raise 'Arg1' if ARGV[0].nil?
+
+input = File.read(ARGV[0]).split.map(&:to_i)
+
+def metas(tree)
+  ch, mt = tree.take 2
+
+  pos = 2
+  res = nil
+
+  if ch.zero?
+    res = tree[pos..-1].take(mt).sum
+  else
+    childs = Array.new(ch).map do
+      child = metas tree[pos..-1]
+      pos += child[1]
+      child[0]
+    end
+
+    res = tree[pos..-1].take(mt).map { |it| childs[it - 1] || 0 }.sum
+  end
+
+  [res, pos + mt]
+end
+
+puts metas(input).first
