view day4.rb @ 21:46fb65f2cb94

Add Day 14 part 1 implementation If you look at columns and see "how far can this go" then the weight is just the sum of the partial triangle numbers of highest_weight to highest_weight minus number of rocks that can roll up to there. Part 2 seems to involve actually moving the rocks, though!
author IBBoard <dev@ibboard.co.uk>
date Sat, 16 Dec 2023 10:29:36 +0000
parents 9da7a71b313d
children
line wrap: on
line source

#! /usr/bin/env ruby

require 'set'

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(chomp: true) do |line|
	winning_number_str, our_number_str = line.split(":")[1].split("|").map(&:chomp)
	winning_numbers = Set.new(winning_number_str.split(" ").map(&:to_i))
	our_numbers = Set.new(our_number_str.split(" ").map(&:to_i))
	our_winning_numbers = winning_numbers & our_numbers
	sum += our_winning_numbers.length > 0 ? 2 ** (our_winning_numbers.length - 1) : 0
end

puts sum