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.
63 lines
1.4 KiB
63 lines
1.4 KiB
tiles = DATA.readlines(chomp: true).flat_map.with_index {|row, y|
|
|
row.chars.filter_map.with_index {|tile, x|
|
|
tile == ?# ? nil : [[y,x], tile]
|
|
}
|
|
}.to_h
|
|
|
|
start, stop = %w[ S E ].map {|t| tiles.find { _2 == t }[0] }
|
|
|
|
scores = Hash.new(Float::INFINITY)
|
|
scores[[start, [0,1]]] = 0
|
|
|
|
queue = [[start, [0,1]]]
|
|
visited = Set.new
|
|
loop do
|
|
fail if queue.empty?
|
|
|
|
current, dir = queue.shift
|
|
visited << [current, dir]
|
|
score = scores.fetch([current, dir])
|
|
break if current == stop
|
|
|
|
# straight ahead
|
|
yx = current.zip(dir).map { _1 + _2 }
|
|
if tiles.has_key?(yx)
|
|
scores[[yx, dir]] = [scores[[yx, dir]], score + 1].min
|
|
queue << [yx, dir] unless visited.include?([yx, dir])
|
|
end
|
|
|
|
# sides
|
|
[-1, 1].each do |d|
|
|
delta = dir.reverse.zip([d, d]).map { _1 * _2 }
|
|
yx = current.zip(delta).map { _1 + _2 }
|
|
if tiles.has_key?(yx)
|
|
scores[[yx, delta]] = [scores[[yx, delta]], score + 1001].min
|
|
queue << [yx, delta] unless visited.include?([yx, delta])
|
|
end
|
|
end
|
|
|
|
queue.uniq!
|
|
queue.sort_by! { scores.fetch(_1) }
|
|
end
|
|
|
|
pp scores.find {|(yx,_),_| yx == stop }[1]
|
|
|
|
__END__
|
|
#################
|
|
#...#...#...#..E#
|
|
#.#.#.#.#.#.#.#^#
|
|
#.#.#.#...#...#^#
|
|
#.#.#.#.###.#.#^#
|
|
#>>v#.#.#.....#^#
|
|
#^#v#.#.#.#####^#
|
|
#^#v..#.#.#>>>>^#
|
|
#^#v#####.#^###.#
|
|
#^#v#..>>>>^#...#
|
|
#^#v###^#####.###
|
|
#^#v#>>^#.....#.#
|
|
#^#v#^#####.###.#
|
|
#^#v#^........#.#
|
|
#^#v#^#########.#
|
|
#S#>>^..........#
|
|
#################
|