diff day2b.rb @ 1:49dd1ae93696

Implement day 2 with map and reduce
author IBBoard <dev@ibboard.co.uk>
date Sat, 02 Dec 2023 16:16:58 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/day2b.rb	Sat Dec 02 16:16:58 2023 +0000
@@ -0,0 +1,19 @@
+#! /usr/bin/env ruby
+
+if ARGV.length != 1
+        abort("Incorrect arguments - needs input file")
+elsif not File.exist? (ARGV[0])
+	abort("File #{ARGV[0]} did not exist")
+end
+
+file = ARGV[0]
+
+sum = 0
+File.open(file, "r").each_line do |line|
+	_, all_picks = line.split(":")
+	sum += all_picks.split(";").map do |pick|
+		pick.split(",").to_h { |colour_pick| colour_pick.strip.split(" ").reverse }.transform_values(&:to_i)
+	end.reduce(Hash.new) {|reduced,obj| reduced.merge(obj) { |key,val1,val2| val1 > val2 ? val1 : val2 }}.values.reduce(1, :*)
+end
+
+puts sum
\ No newline at end of file