diff day6b.rb @ 7:6e0615e54e71

Restructure input parsing for day6b The rest of the algorithm is the same
author IBBoard <dev@ibboard.co.uk>
date Thu, 07 Dec 2023 08:23:04 +0000
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/day6b.rb	Thu Dec 07 08:23:04 2023 +0000
@@ -0,0 +1,21 @@
+#! /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]
+
+time, distance = File.open(file, "r").each_line(chomp: true).map {|v| v.split(":")[1].split(" ").join("").to_i}
+events = [[time, distance]]
+
+product = events.map do |event|
+	mid = (event[0].to_f / 2).floor
+	first_success = (1..mid).bsearch {|i| (event[0] - i) * i > event[1]}
+	num_success = (mid - first_success + 1) * 2
+	event[0].odd? ? num_success : num_success - 1
+end.reduce(:*)
+
+puts product
\ No newline at end of file