You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
70 lines
1.3 KiB
70 lines
1.3 KiB
Part = Data.define(:number) do
|
|
def eql?(other)
|
|
self.object_id == other.object_id
|
|
end
|
|
end
|
|
CurrentNumber = Data.define(:coords, :acc)
|
|
|
|
cur_num = nil
|
|
nums = {}
|
|
syms = {}
|
|
ARGF.readlines(chomp: true).each.with_index do |row, y|
|
|
row.chars.each.with_index do |c, x|
|
|
case c
|
|
when /\d/
|
|
if cur_num
|
|
cur_num.coords << [y,x]
|
|
cur_num.acc << c
|
|
else
|
|
cur_num = CurrentNumber.new([[y,x]], c)
|
|
end
|
|
else
|
|
if cur_num
|
|
part = Part.new(cur_num.acc.to_i)
|
|
nums.merge!(cur_num.coords.to_h { [_1, part] })
|
|
cur_num = nil
|
|
end
|
|
|
|
if c != ?.
|
|
syms[[y,x]] = c
|
|
end
|
|
end
|
|
end
|
|
|
|
if cur_num
|
|
part = Part.new(cur_num.acc.to_i)
|
|
nums.merge!(cur_num.coords.to_h { [_1, part] })
|
|
cur_num = nil
|
|
end
|
|
end
|
|
|
|
# part one
|
|
p syms
|
|
.flat_map {|(y,x), _|
|
|
dy = (-1..1).to_a
|
|
dx = (-1..1).to_a
|
|
dy.product(dx)
|
|
.map {|dy,dx| [y+dy,x+dx] }
|
|
.filter_map { nums.fetch(_1, nil) }
|
|
.uniq
|
|
}
|
|
.sum(&:number)
|
|
|
|
p syms
|
|
.select { _2 == ?* }
|
|
.filter_map {|(y,x), _|
|
|
dy = (-1..1).to_a
|
|
dx = (-1..1).to_a
|
|
parts = dy.product(dx)
|
|
.map {|dy,dx| [y+dy,x+dx] }
|
|
.filter_map { nums.fetch(_1, nil) }
|
|
.uniq
|
|
|
|
if parts.length == 2
|
|
parts
|
|
else
|
|
nil
|
|
end
|
|
}
|
|
.sum { _1.map(&:number).inject(:*) }
|