Day twenty-one Part one
This commit is contained in:
parent
c6c314ca30
commit
3e6096cb9f
2 changed files with 342 additions and 0 deletions
5
21/input
Normal file
5
21/input
Normal file
|
@ -0,0 +1,5 @@
|
||||||
|
140A
|
||||||
|
169A
|
||||||
|
170A
|
||||||
|
528A
|
||||||
|
340A
|
337
21/main.rb
Executable file
337
21/main.rb
Executable file
|
@ -0,0 +1,337 @@
|
||||||
|
#!/usr/bin/env ruby
|
||||||
|
|
||||||
|
require "debug"
|
||||||
|
|
||||||
|
input = (ARGV.first.nil? ? DATA : ARGF)
|
||||||
|
.readlines(chomp: true)
|
||||||
|
.map(&:chars)
|
||||||
|
|
||||||
|
@arrow_dirs = {
|
||||||
|
"<" => [0, -1],
|
||||||
|
"^" => [-1, 0],
|
||||||
|
">" => [0, 1],
|
||||||
|
"v" => [1, 0],
|
||||||
|
}
|
||||||
|
|
||||||
|
@dir_arrows = @arrow_dirs.invert
|
||||||
|
|
||||||
|
numeric_movements = {
|
||||||
|
"0" => ["^", ">"],
|
||||||
|
"A" => ["<", "^"],
|
||||||
|
"1" => ["^", ">"],
|
||||||
|
"2" => ["<", "^", ">", "v"],
|
||||||
|
"3" => ["<", "^", "v"],
|
||||||
|
"4" => ["^", ">", "v"],
|
||||||
|
"5" => ["<", "^", ">", "v"],
|
||||||
|
"6" => ["<", "^", "v"],
|
||||||
|
"7" => [">", "v"],
|
||||||
|
"8" => ["<", ">", "v"],
|
||||||
|
"9" => ["<", "v"],
|
||||||
|
}
|
||||||
|
|
||||||
|
directional_movements = {
|
||||||
|
"^" => [">", "v"],
|
||||||
|
"A" => ["v", "<"],
|
||||||
|
"<" => [">"],
|
||||||
|
"v" => ["^", ">", "<"],
|
||||||
|
">" => ["^", "<"],
|
||||||
|
}
|
||||||
|
|
||||||
|
numeric_keypad = [
|
||||||
|
["7", "8", "9"],
|
||||||
|
["4", "5", "6"],
|
||||||
|
["1", "2", "3"],
|
||||||
|
[nil, "0", "A"],
|
||||||
|
]
|
||||||
|
|
||||||
|
directional_keypad = [
|
||||||
|
[nil, "^", "A"],
|
||||||
|
["<", "v", ">"],
|
||||||
|
]
|
||||||
|
|
||||||
|
class PriorityQueue < Array
|
||||||
|
alias_method :extract_min, :shift
|
||||||
|
|
||||||
|
def <<(v)
|
||||||
|
super
|
||||||
|
|
||||||
|
sort_by!(&:last)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def discover(char, graph)
|
||||||
|
(0...graph.size).each do |x|
|
||||||
|
(0...graph.first.size).each do |y|
|
||||||
|
return [x, y] if char == graph.dig(x, y)
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def dijkstra_paths(prev, target, current_path = [])
|
||||||
|
return [current_path.reverse] if prev[target].nil?
|
||||||
|
|
||||||
|
prev[target].flat_map do |parent|
|
||||||
|
dijkstra_paths(prev, parent, current_path + [target])
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
def dijkstra_with_lookup(graph, movements, start_char, target_char)
|
||||||
|
start = discover(start_char, graph)
|
||||||
|
target = discover(target_char, graph)
|
||||||
|
|
||||||
|
dijkstra(graph, movements, start, target)
|
||||||
|
end
|
||||||
|
|
||||||
|
def dijkstra(graph, movements, start, target)
|
||||||
|
queue = PriorityQueue.new
|
||||||
|
dest = Hash.new
|
||||||
|
prev = Hash.new
|
||||||
|
|
||||||
|
queue << [start, 0]
|
||||||
|
dest[start] = 0
|
||||||
|
prev[start] = nil
|
||||||
|
|
||||||
|
until queue.empty?
|
||||||
|
node, moves = queue.extract_min
|
||||||
|
|
||||||
|
movements[graph.dig(*node)]
|
||||||
|
.map { @arrow_dirs[_1] }
|
||||||
|
.map { |dx, dy| [node.first + dx, node.last + dy] }
|
||||||
|
.each do |neighbor|
|
||||||
|
new_moves = moves + 1
|
||||||
|
if dest[neighbor].nil? || new_moves < dest[neighbor]
|
||||||
|
dest[neighbor] = new_moves
|
||||||
|
prev[neighbor] = [node]
|
||||||
|
queue << [neighbor, new_moves]
|
||||||
|
elsif new_moves == dest[neighbor]
|
||||||
|
prev[neighbor] << node
|
||||||
|
end
|
||||||
|
end
|
||||||
|
end
|
||||||
|
|
||||||
|
paths = dijkstra_paths(prev, target)
|
||||||
|
|
||||||
|
return dest, paths
|
||||||
|
end
|
||||||
|
|
||||||
|
# calculate most efficient number pad paths
|
||||||
|
# actually AN efficient path; ties not included :\
|
||||||
|
# @numeric_map = numeric_keypad
|
||||||
|
# .flatten.reject { _1.nil? }
|
||||||
|
# .permutation(2).to_a
|
||||||
|
# .map { [[_1, _2], dijkstra_with_lookup(numeric_keypad, numeric_movements, _1, _2)] }
|
||||||
|
# .map do |(start, target), (dest, paths)|
|
||||||
|
# presses = paths.map do |path|
|
||||||
|
# ([discover(start, numeric_keypad)] + path)
|
||||||
|
# .each_cons(2).to_a
|
||||||
|
# .map { @dir_arrows[[_2.first - _1.first, _2.last - _1.last]] }.join
|
||||||
|
# end
|
||||||
|
# [[start, target], presses]
|
||||||
|
# end
|
||||||
|
# .to_h
|
||||||
|
|
||||||
|
@numeric_map = {
|
||||||
|
["7", "8"]=>[">"],
|
||||||
|
["7", "9"]=>[">>"],
|
||||||
|
["7", "4"]=>["v"],
|
||||||
|
["7", "5"]=>["v>"],
|
||||||
|
["7", "6"]=>["v>>"],
|
||||||
|
["7", "1"]=>["vv"],
|
||||||
|
["7", "2"]=>["vv>"],
|
||||||
|
["7", "3"]=>["vv>>"],
|
||||||
|
["7", "0"]=>[">vvv", "v>vv", "vv>v"],
|
||||||
|
["7", "A"]=>[">>vvv"],
|
||||||
|
["8", "7"]=>["<"],
|
||||||
|
["8", "9"]=>[">"],
|
||||||
|
["8", "4"]=>["<v", "v<"],
|
||||||
|
["8", "5"]=>["v"],
|
||||||
|
["8", "6"]=>["v>"],
|
||||||
|
["8", "1"]=>["<vv", "vv<"],
|
||||||
|
["8", "2"]=>["vv"],
|
||||||
|
["8", "3"]=>["vv>"],
|
||||||
|
["8", "0"]=>["vvv"],
|
||||||
|
["8", "A"]=>["vvv>"],
|
||||||
|
["9", "7"]=>["<<"],
|
||||||
|
["9", "8"]=>["<"],
|
||||||
|
["9", "4"]=>["<<v"],
|
||||||
|
["9", "5"]=>["<v", "v<"],
|
||||||
|
["9", "6"]=>["v"],
|
||||||
|
["9", "1"]=>["<<vv", "vv<<"],
|
||||||
|
["9", "2"]=>["<vv", "vv<"],
|
||||||
|
["9", "3"]=>["vv"],
|
||||||
|
["9", "0"]=>["<vvv", "vvv<"],
|
||||||
|
["9", "A"]=>["vvv"],
|
||||||
|
["4", "7"]=>["^"],
|
||||||
|
["4", "8"]=>["^>", ">^"],
|
||||||
|
["4", "9"]=>["^>>", ">>^"],
|
||||||
|
["4", "5"]=>[">"],
|
||||||
|
["4", "6"]=>[">>"],
|
||||||
|
["4", "1"]=>["v"],
|
||||||
|
["4", "2"]=>["v>"],
|
||||||
|
["4", "3"]=>["v>>"],
|
||||||
|
["4", "0"]=>[">vv"],
|
||||||
|
["4", "A"]=>[">>vv"],
|
||||||
|
["5", "7"]=>["<^", "^<"],
|
||||||
|
["5", "8"]=>["^"],
|
||||||
|
["5", "9"]=>["^>", ">^"],
|
||||||
|
["5", "4"]=>["<"],
|
||||||
|
["5", "6"]=>[">"],
|
||||||
|
["5", "1"]=>["<v", "v<"],
|
||||||
|
["5", "2"]=>["v"],
|
||||||
|
["5", "3"]=>["v>"],
|
||||||
|
["5", "0"]=>["vv"],
|
||||||
|
["5", "A"]=>["vv>"],
|
||||||
|
["6", "7"]=>["<<^", "^<<"],
|
||||||
|
["6", "8"]=>["<^", "^<"],
|
||||||
|
["6", "9"]=>["^"],
|
||||||
|
["6", "4"]=>["<<"],
|
||||||
|
["6", "5"]=>["<"],
|
||||||
|
["6", "1"]=>["<<v", "v<<"],
|
||||||
|
["6", "2"]=>["<v", "v<"],
|
||||||
|
["6", "3"]=>["v"],
|
||||||
|
["6", "0"]=>["<vv", "vv<"],
|
||||||
|
["6", "A"]=>["vv"],
|
||||||
|
["1", "7"]=>["^^"],
|
||||||
|
["1", "8"]=>["^^>", ">^^"],
|
||||||
|
["1", "9"]=>["^^>>", ">>^^"],
|
||||||
|
["1", "4"]=>["^"],
|
||||||
|
["1", "5"]=>["^>", ">^"],
|
||||||
|
["1", "6"]=>["^>>", ">>^"],
|
||||||
|
["1", "2"]=>[">"],
|
||||||
|
["1", "3"]=>[">>"],
|
||||||
|
["1", "0"]=>[">v"],
|
||||||
|
["1", "A"]=>[">>v", ">v>"],
|
||||||
|
["2", "7"]=>["<^^", "^^<"],
|
||||||
|
["2", "8"]=>["^^"],
|
||||||
|
["2", "9"]=>["^^>", ">^^"],
|
||||||
|
["2", "4"]=>["<^", "^<"],
|
||||||
|
["2", "5"]=>["^"],
|
||||||
|
["2", "6"]=>["^>", ">^"],
|
||||||
|
["2", "1"]=>["<"],
|
||||||
|
["2", "3"]=>[">"],
|
||||||
|
["2", "0"]=>["v"],
|
||||||
|
["2", "A"]=>[">v", "v>"],
|
||||||
|
["3", "7"]=>["<<^^", "^^<<"],
|
||||||
|
["3", "8"]=>["<^^", "^^<"],
|
||||||
|
["3", "9"]=>["^^"],
|
||||||
|
["3", "4"]=>["<<^", "^<<"],
|
||||||
|
["3", "5"]=>["<^", "^<"],
|
||||||
|
["3", "6"]=>["^"],
|
||||||
|
["3", "1"]=>["<<"],
|
||||||
|
["3", "2"]=>["<"],
|
||||||
|
["3", "0"]=>["<v", "v<"],
|
||||||
|
["3", "A"]=>["v"],
|
||||||
|
["0", "7"]=>["^^^<"],
|
||||||
|
["0", "8"]=>["^^^"],
|
||||||
|
["0", "9"]=>["^^^>", ">^^^"],
|
||||||
|
["0", "4"]=>["^^<"],
|
||||||
|
["0", "5"]=>["^^"],
|
||||||
|
["0", "6"]=>["^^>", ">^^"],
|
||||||
|
["0", "1"]=>["^<"],
|
||||||
|
["0", "2"]=>["^"],
|
||||||
|
["0", "3"]=>["^>", ">^"],
|
||||||
|
["0", "A"]=>[">"],
|
||||||
|
["A", "7"]=>["^^^<<"],
|
||||||
|
["A", "8"]=>["<^^^", "^^^<"],
|
||||||
|
["A", "9"]=>["^^^"],
|
||||||
|
["A", "4"]=>["^^<<"],
|
||||||
|
["A", "5"]=>["<^^", "^^<"],
|
||||||
|
["A", "6"]=>["^^"],
|
||||||
|
["A", "1"]=>["^<<"],
|
||||||
|
["A", "2"]=>["<^", "^<"],
|
||||||
|
["A", "3"]=>["^"],
|
||||||
|
["A", "0"]=>["<"]
|
||||||
|
}
|
||||||
|
|
||||||
|
# cache efficient direction movements
|
||||||
|
# @directional_map = directional_keypad
|
||||||
|
# .flatten.reject { _1.nil? }
|
||||||
|
# .permutation(2).to_a
|
||||||
|
# .map { [[_1, _2], dijkstra_with_lookup(directional_keypad, directional_movements, _1, _2)] }
|
||||||
|
# .map do |(start, target), (dest, paths)|
|
||||||
|
# presses = paths.map do |path|
|
||||||
|
# ([discover(start, directional_keypad)] + path)
|
||||||
|
# .each_cons(2).to_a
|
||||||
|
# .map { @dir_arrows[[_2.first - _1.first, _2.last - _1.last]] }.join
|
||||||
|
# end
|
||||||
|
# [[start, target], presses]
|
||||||
|
# end
|
||||||
|
# .to_h
|
||||||
|
|
||||||
|
@directional_map = {
|
||||||
|
["^", "A"]=>[">"],
|
||||||
|
["^", "<"]=>["v<"],
|
||||||
|
["^", "v"]=>["v"],
|
||||||
|
["^", ">"]=>[">v", "v>"],
|
||||||
|
["A", "^"]=>["<"],
|
||||||
|
["A", "<"]=>["v<<", "<v<"],
|
||||||
|
["A", "v"]=>["v<", "<v"],
|
||||||
|
["A", ">"]=>["v"],
|
||||||
|
["<", "^"]=>[">^"],
|
||||||
|
["<", "A"]=>[">>^"],
|
||||||
|
["<", "v"]=>[">"],
|
||||||
|
["<", ">"]=>[">>"],
|
||||||
|
["v", "^"]=>["^"],
|
||||||
|
["v", "A"]=>["^>", ">^"],
|
||||||
|
["v", "<"]=>["<"],
|
||||||
|
["v", ">"]=>[">"],
|
||||||
|
[">", "^"]=>["^<", "<^"],
|
||||||
|
[">", "A"]=>["^"],
|
||||||
|
[">", "<"]=>["<<"],
|
||||||
|
[">", "v"]=>["<"]
|
||||||
|
}
|
||||||
|
|
||||||
|
def solve(input, times = 3)
|
||||||
|
@robot_1 = input
|
||||||
|
@robot_2 = @robot_1
|
||||||
|
.map { ["A"] + _1 }
|
||||||
|
.map { _1.each_cons(2).to_a }
|
||||||
|
.map { _1.map { |move| @numeric_map[move].first }.join("A") }
|
||||||
|
.map { _1 + "A" }
|
||||||
|
|
||||||
|
(3..times).each do |num|
|
||||||
|
prev_robot = instance_variable_get(:"@robot_#{num - 1}")
|
||||||
|
|
||||||
|
robot = prev_robot
|
||||||
|
.map { _1.chars }
|
||||||
|
.map { ["A"] + _1 }
|
||||||
|
.map { _1.each_cons(2).to_a }
|
||||||
|
.map do
|
||||||
|
_1.map do |move|
|
||||||
|
@directional_map[move].nil? ? "" : @directional_map[move].first
|
||||||
|
end.join("A")
|
||||||
|
end
|
||||||
|
.map { _1 + "A" }
|
||||||
|
instance_variable_set(:"@robot_#{num}", robot)
|
||||||
|
|
||||||
|
@last_num = num
|
||||||
|
end
|
||||||
|
|
||||||
|
last_robot = instance_variable_get(:"@robot_#{@last_num}")
|
||||||
|
me = last_robot
|
||||||
|
.map { _1.chars }
|
||||||
|
.map { ["A"] + _1 }
|
||||||
|
.map { _1.each_cons(2).to_a }
|
||||||
|
.map do
|
||||||
|
_1.map do |move|
|
||||||
|
@directional_map[move].nil? ? "" : @directional_map[move].first
|
||||||
|
end.join("A")
|
||||||
|
end
|
||||||
|
.map { _1 + "A" }
|
||||||
|
|
||||||
|
me
|
||||||
|
end
|
||||||
|
|
||||||
|
me = solve(input, 3)
|
||||||
|
|
||||||
|
me.each_with_index do |entry, i|
|
||||||
|
p "#{entry.length} * #{input[i].join.to_i} = #{entry.length * input[i].join.to_i}"
|
||||||
|
end
|
||||||
|
p me.map.with_index { |entry, i| entry.length * input[i].join.to_i }.sum
|
||||||
|
|
||||||
|
__END__
|
||||||
|
029A
|
||||||
|
980A
|
||||||
|
179A
|
||||||
|
456A
|
||||||
|
379A
|
Loading…
Reference in a new issue