Advent of Code - Day 12: Passage Pathing

lib/advent_of_code/2021/day-12.livemd

Tyler Johnson

@tylerj

aoc

Share to X

Share to Bluesky

More notebooks

Advent of Code - Day 12: Passage Pathing

Mix.install([:kino])

Day 12: Passage Pathing

Part One: Prompt

With your submarine's subterranean subsystems subsisting suboptimally, the only way you're getting out of this cave anytime soon is by finding a path yourself. Not just a path - the only way to know if you've found the best path is to find all of them.

Fortunately, the sensors are still mostly working, and so you build a rough map of the remaining caves (your puzzle input). For example:

start-A
start-b
A-c
A-b
b-d
A-end
b-end

This is a list of how all of the caves are connected. You start in the cave named start, and your destination is the cave named end. An entry like b-d means that cave b is connected to cave d - that is, you can move between them.

So, the above cave system looks roughly like this:

    start
    /   \
c--A-----b--d
    \   /
     end

Your goal is to find the number of distinct paths that start at start, end at end, and don't visit small caves more than once. There are two types of caves: big caves (written in uppercase, like A) and small caves (written in lowercase, like b). It would be a waste of time to visit any small cave more than once, but big caves are large enough that it might be worth visiting them multiple times. So, all paths you find should visit small caves at most once, and can visit big caves any number of times.

Given these rules, there are 10 paths through this example cave system:

start,A,b,A,c,A,end
start,A,b,A,end
start,A,b,end
start,A,c,A,b,A,end
start,A,c,A,b,end
start,A,c,A,end
start,A,end
start,b,A,c,A,end
start,b,A,end
start,b,end

(Each line in the above list corresponds to a single path; the caves visited by that path are listed in the order they are visited and separated by commas.)

Note that in this cave system, cave d is never visited by any path: to do so, cave b would need to be visited twice (once on the way to cave d and a second time when returning from cave d), and since cave b is small, this is not allowed.

Here is a slightly larger example:

dc-end
HN-start
start-kj
dc-start
dc-HN
LN-dc
HN-end
kj-sa
kj-HN
kj-dc

The 19 paths through it are as follows:

start,HN,dc,HN,end
start,HN,dc,HN,kj,HN,end
start,HN,dc,end
start,HN,dc,kj,HN,end
start,HN,end
start,HN,kj,HN,dc,HN,end
start,HN,kj,HN,dc,end
start,HN,kj,HN,end
start,HN,kj,dc,HN,end
start,HN,kj,dc,end
start,dc,HN,end
start,dc,HN,kj,HN,end
start,dc,end
start,dc,kj,HN,end
start,kj,HN,dc,HN,end
start,kj,HN,dc,end
start,kj,HN,end
start,kj,dc,HN,end
start,kj,dc,end

Finally, this even larger example has 226 paths through it:

fs-end
he-DX
fs-he
start-DX
pj-DX
end-zg
zg-sl
zg-pj
pj-he
RW-he
fs-DX
pj-RW
zg-RW
start-pj
he-WI
zg-he
pj-fs
start-RW

How many paths through this cave system are there that visit small caves at most once?

To begin, get your puzzle input.

Part One: Solution

input = Kino.Input.textarea("Please paste your input file:")

caves =
  input
  |> Kino.Input.read()
  |> String.split("\n", trim: true)
  |> Enum.map(&String.split(&1, "-"))
  |> Enum.reduce(%{}, fn [a, b], map ->
    map
    |> Map.update(a, [b], &[b | &1])
    |> Map.update(b, [a], &[a | &1])
  end)

%{
  "FN" => ["li", "kc", "end", "qy"],
  "XQ" => ["nv"],
  "ZP" => ["qy", "start", "wx", "yc", "li", "kc"],
  "end" => ["nv", "qy", "FN"],
  "kc" => ["nv", "li", "wx", "FN", "ZP", "qy"],
  "li" => ["kc", "wx", "FN", "yc", "qy", "ZP"],
  "nv" => ["XQ", "kc", "end", "xr", "qy"],
  "qy" => ["ZP", "nv", "li", "end", "FN", "kc"],
  "start" => ["wx", "ZP", "yc"],
  "wx" => ["start", "kc", "li", "yc", "ZP"],
  "xr" => ["nv"],
  "yc" => ["wx", "li", "ZP", "start"]
}

defmodule Caves do
  @cave_types [
    {~r/\Astart\z/i, :start},
    {~r/\Aend\z/i, :end},
    {~r/\A[a-z]+\z/, :small},
    {~r/\A[A-Z]+\z/, :big}
  ]
  def cave_type(cave) do
    with {_regex, type} <-
           Enum.find(
             @cave_types,
             fn {regex, _} -> String.match?(cave, regex) end
           ) do
      type
    end
  end

  def find_paths(map) do
    visit_cave(
      map,
      "start",
      {"start"},
      MapSet.new(["start"])
    )
  end

  def visit_cave(map, cave, path, visited) do
    Enum.map(map[cave], fn neighbor ->
      cond do
        neighbor == "end" ->
          Tuple.append(path, "end")

        neighbor in visited ->
          nil

        cave_type(neighbor) == :small ->
          visit_cave(map, neighbor, Tuple.append(path, neighbor), MapSet.put(visited, neighbor))

        true ->
          visit_cave(map, neighbor, Tuple.append(path, neighbor), visited)
      end
    end)
  end
end

{:module, Caves, <<70, 79, 82, 49, 0, 0, 14, ...>>, {:visit_cave, 4}}

paths =
  Caves.find_paths(caves)
  |> List.flatten()
  |> Enum.filter(& &1)

[
  {"start", "wx", "kc", "nv", "end"},
  {"start", "wx", "kc", "nv", "qy", "ZP", "yc", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "qy", "ZP", "yc", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "qy", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "qy", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "qy", "end"},
  {"start", "wx", "kc", "nv", "qy", "FN", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "qy", "FN", "end"},
  {"start", "wx", "kc", "li", "FN", "end"},
  {"start", "wx", "kc", "li", "FN", "qy", "nv", "end"},
  {"start", "wx", "kc", "li", "FN", "qy", "end"},
  {"start", "wx", "kc", "li", "FN", "qy", "FN", "end"},
  {"start", "wx", "kc", "li", "yc", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "li", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "li", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "li", "qy", "nv", "end"},
  {"start", "wx", "kc", "li", "qy", "end"},
  {"start", "wx", "kc", "li", "qy", "FN", "end"},
  {"start", "wx", "kc", "li", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "li", "ZP", "qy", "end"},
  {"start", "wx", "kc", "li", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "li", "ZP", "yc", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "li", "ZP", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "li", "ZP", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "FN", "qy", "nv", "end"},
  {"start", "wx", "kc", "FN", "li", "FN", "qy", "end"},
  {"start", "wx", "kc", "FN", "li", "FN", "qy", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "yc", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "FN", "li", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "FN", "li", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "qy", "nv", "end"},
  {"start", "wx", "kc", "FN", "li", "qy", "end"},
  {"start", "wx", "kc", "FN", "li", "qy", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "qy", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "yc", "ZP", "qy", "nv", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "FN", "li", "ZP", "yc", "ZP", "qy", "FN", ...},
  {"start", "wx", "kc", "FN", "end"},
  {"start", "wx", "kc", "FN", "qy", "ZP", "yc", "li", ...},
  {"start", "wx", "kc", "FN", "qy", "ZP", "yc", ...},
  {"start", "wx", "kc", "FN", "qy", "ZP", ...},
  {"start", "wx", "kc", "FN", "qy", ...},
  {"start", "wx", "kc", "FN", ...},
  {"start", "wx", "kc", ...},
  {"start", "wx", ...},
  {"start", ...},
  {...},
  ...
]

length(paths)

How many paths through this cave system are there that visit small caves at most once?

Your puzzle answer was `5874`.

The first half of this puzzle is complete! It provides one gold star: *

Part Two: Prompt

After reviewing the available paths, you realize you might have time to visit a single small cave twice. Specifically, big caves can be visited any number of times, a single small cave can be visited at most twice, and the remaining small caves can be visited at most once. However, the caves named start and end can only be visited exactly once each: once you leave the start cave, you may not return to it, and once you reach the end cave, the path must end immediately.

Now, the 36 possible paths through the first example above are:

start,A,b,A,b,A,c,A,end
start,A,b,A,b,A,end
start,A,b,A,b,end
start,A,b,A,c,A,b,A,end
start,A,b,A,c,A,b,end
start,A,b,A,c,A,c,A,end
start,A,b,A,c,A,end
start,A,b,A,end
start,A,b,d,b,A,c,A,end
start,A,b,d,b,A,end
start,A,b,d,b,end
start,A,b,end
start,A,c,A,b,A,b,A,end
start,A,c,A,b,A,b,end
start,A,c,A,b,A,c,A,end
start,A,c,A,b,A,end
start,A,c,A,b,d,b,A,end
start,A,c,A,b,d,b,end
start,A,c,A,b,end
start,A,c,A,c,A,b,A,end
start,A,c,A,c,A,b,end
start,A,c,A,c,A,end
start,A,c,A,end
start,A,end
start,b,A,b,A,c,A,end
start,b,A,b,A,end
start,b,A,b,end
start,b,A,c,A,b,A,end
start,b,A,c,A,b,end
start,b,A,c,A,c,A,end
start,b,A,c,A,end
start,b,A,end
start,b,d,b,A,c,A,end
start,b,d,b,A,end
start,b,d,b,end
start,b,end

The slightly larger example above now has 103 paths through it, and the even larger example now has 3509 paths through it.

Given these new rules, how many paths through this cave system are there?

Part Two: Solution

defmodule Caves2 do
  defdelegate cave_type(cave), to: Caves

  def find_paths(map) do
    visit_cave(
      map,
      "start",
      {"start"}
    )
  end

  def visit_cave(map, cave, path, visited \\ MapSet.new(), visited_2x \\ false) do
    Enum.map(map[cave], fn neighbor ->
      cond do
        neighbor == "start" ->
          nil

        neighbor == "end" ->
          Tuple.append(path, "end")

        neighbor in visited && visited_2x ->
          nil

        neighbor in visited ->
          visit_cave(
            map,
            neighbor,
            Tuple.append(path, neighbor),
            visited,
            true
          )

        cave_type(neighbor) == :small ->
          visit_cave(
            map,
            neighbor,
            Tuple.append(path, neighbor),
            MapSet.put(visited, neighbor),
            visited_2x
          )

        true ->
          visit_cave(
            map,
            neighbor,
            Tuple.append(path, neighbor),
            visited,
            visited_2x
          )
      end
    end)
  end
end

{:module, Caves2, <<70, 79, 82, 49, 0, 0, 13, ...>>, {:visit_cave, 5}}

paths =
  Caves2.find_paths(caves)
  |> List.flatten()
  |> Enum.filter(& &1)

[
  {"start", "wx", "kc", "nv", "XQ", "nv", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "ZP", "yc", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "ZP", "yc", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "FN", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "XQ", "nv", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "FN", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "FN", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "ZP", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "li", "ZP", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "FN", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "FN", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "ZP", "yc", "ZP", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "li", "ZP", "yc", "ZP", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "ZP", "yc", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "ZP", "yc", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "ZP", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "FN", "li", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "FN", "qy", "FN", "end"},
  {"start", "wx", "kc", "nv", "kc", "ZP", "qy", "ZP", "yc", "li", "FN", ...},
  {"start", "wx", "kc", "nv", "kc", "ZP", "qy", "ZP", "yc", "ZP", ...},
  {"start", "wx", "kc", "nv", "kc", "ZP", "qy", "ZP", "li", ...},
  {"start", "wx", "kc", "nv", "kc", "ZP", "qy", "li", ...},
  {"start", "wx", "kc", "nv", "kc", "ZP", "qy", ...},
  {"start", "wx", "kc", "nv", "kc", "ZP", ...},
  {"start", "wx", "kc", "nv", "kc", ...},
  {"start", "wx", "kc", "nv", ...},
  {"start", "wx", "kc", ...},
  {"start", "wx", ...},
  {"start", ...},
  {...},
  ...
]