Advent of Code 2021 in Julia – Day 12: Passage Pathing

Table of Contents

The puzzle today competely wrecked me! I spent like three hours on this one.

Part 1

Show challenge - day 12, part 1

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?

Here’s the provided input.

Show input - day 12
LA-sn
LA-mo
LA-zs
end-RD
sn-mo
end-zs
vx-start
mh-mo
mh-start
zs-JI
JQ-mo
zs-mo
start-JQ
rk-zs
mh-sn
mh-JQ
RD-mo
zs-JQ
vx-sn
RD-sn
vx-mh
JQ-vx
LA-end
JQ-sn

Here’s the solution for part 1:

using Pipe: @pipe

mutable struct Cave
    label::String
    type::String
    paths::Vector{String}
end

function define_cave(s)
    if s == "start"
        return Cave("start", "start", [])
    elseif s == "end"
        return Cave("end", "end", [])
    elseif isuppercase(s[1])
        return Cave(s, "big", [])
    else
        return Cave(s, "small", [])
    end
end

function define_path!(s1, s2)
    if s2  ["start"]
        push!(caves[getproperty.(caves, :label) .== s1][1].paths,
              caves[getproperty.(caves, :label) .== s2][1].label)
    end
    if s1  ["start"]
        push!(caves[getproperty.(caves, :label) .== s2][1].paths,
            caves[getproperty.(caves, :label) .== s1][1].label)
    end
end

input = @pipe split(strip(input), "\n") .|> string .|>
    split(_, "-") .|> map(string, _)

caves = @pipe input .|> collect |> vcat(_...) |> unique |> map(define_cave, _)
@pipe input .|> define_path!(_...)

dead_ends =
    caves[(@pipe caves .|> getproperty(_, :paths) .|> length .|> ==(1)) .&&
    (@pipe caves .|> getproperty(_, :paths) .|> islowercase(_[1][1])) .&&
    (@pipe caves .|> getproperty(_, :type) .|> ==("small"))]
dead_ends = getproperty.(dead_ends, :label)

function next_step(path)
    out = []
    directions = caves[getproperty.(caves, :label) .== path[end]][1].paths
    for d in directions
        if d  [dead_ends;
                path[begin:end-1][@pipe path[begin:end-1] .|> islowercase(_[1])]]
            continue
        else
            out = [out; [[path; [d]]]]
        end
    end
    return out
end

paths = [["start"]]

while any(last.(paths) .!= "end")
    end_paths = [p for p in paths if p[end] == "end"]
    unfinished_paths = [p for p in paths if p[end] != "end"]
    paths = [end_paths; next_step.(unfinished_paths)...]
end

length(paths) |> print

Part 2

Show challenge - day 12, part 2

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?

Here’s the solution for part 2:

using Pipe: @pipe

mutable struct Cave
    label::String
    type::String
    paths::Vector{String}
end

function define_cave(s)
    if s == "start"
        return Cave("start", "start", [])
    elseif s == "end"
        return Cave("end", "end", [])
    elseif isuppercase(s[1])
        return Cave(s, "big", [])
    else
        return Cave(s, "small", [])
    end
end

function define_path!(s1, s2)
    if s2  ["start"]
        push!(caves[getproperty.(caves, :label) .== s1][1].paths,
              caves[getproperty.(caves, :label) .== s2][1].label)
    end
    if s1  ["start"]
        push!(caves[getproperty.(caves, :label) .== s2][1].paths,
            caves[getproperty.(caves, :label) .== s1][1].label)
    end
end

function next_step(path)
    out = []
    directions = caves[getproperty.(caves, :label) .== path[end]][1].paths
    small = path[@pipe path .|> islowercase(_[1])]
    tally = [sum(small .== i) => i for i in unique(small)]
    if count(s -> s.first == 2, tally) < 1
        small = [i.second for i in tally if i.first == 2]
    end
    for d in directions
        if d  small
            continue
        else
            out = [out; [[path; [d]]]]
        end
    end
    return out
end

input = @pipe split(strip(input), "\n") .|> string .|>
    split(_, "-") .|> map(string, _)

caves = @pipe input .|> collect |> vcat(_...) |> unique |> map(define_cave, _)
@pipe input .|> define_path!(_...)

paths = [["start"]]

path_count = 0
path_count = path_count + sum(last.(paths) .== "end")

while any(last.(paths) .!= "end")
    unfinished_paths = [p for p in paths if p[end] != "end"]
    paths = [next_step.(unfinished_paths)...;]
    path_count = path_count + sum(last.(paths) .== "end")
end

path_count |> print