···
4
+
Mix.install([:kino_aoc])
9
+
<!-- livebook:{"attrs":"eyJhc3NpZ25fdG8iOiJwdXp6bGVfaW5wdXQiLCJkYXkiOiI4Iiwic2Vzc2lvbl9zZWNyZXQiOiJBRFZFTlRfT0ZfQ09ERV9TRVNTSU9OIiwieWVhciI6IjIwMjUifQ","chunks":null,"kind":"Elixir.KinoAOC.HelperCell","livebook_object":"smart_cell"} -->
12
+
{:ok, puzzle_input} =
13
+
KinoAOC.download_puzzle("2025", "8", System.fetch_env!("LB_ADVENT_OF_CODE_SESSION"))
20
+
|> Enum.map(fn raw ->
22
+
|> String.split(",")
23
+
|> Enum.map(&String.to_integer/1)
32
+
defmodule JunctionBoxes do
33
+
def all_pairs([]), do: []
35
+
def all_pairs([x | rest]) do
36
+
for(y <- rest, do: {x, y}) ++ all_pairs(rest)
39
+
def dist2({ax, ay, az}, {bx, by, bz}) do
44
+
dx ** 2 + dy ** 2 + dz ** 2
47
+
def group([], {a, b}), do: [MapSet.new([a, b])]
48
+
def group([set | rest], {a, b}) do
50
+
a in set and b in set -> [set | rest]
51
+
a in set or b in set -> set |> MapSet.put(a) |> MapSet.put(b) |> do_squash(rest, a, b, [])
52
+
true -> [set | group(rest, {a, b})]
56
+
defp do_squash(curr, [], _, _, acc), do: [curr | acc]
57
+
defp do_squash(curr, [x | rest], a, b, acc) do
58
+
if a in x or b in x do
59
+
[MapSet.union(curr, x) | acc ++ rest]
61
+
do_squash(curr, rest, a, b, [x | acc])
69
+
JunctionBoxes.all_pairs(boxes)
70
+
|> Enum.sort_by(fn {a, b} -> JunctionBoxes.dist2(a, b) end)
73
+
<!-- livebook:{"branch_parent_index":1} -->
80
+
|> Enum.reduce([], &JunctionBoxes.group(&2, &1))
81
+
|> Enum.map(&MapSet.size/1)
87
+
<!-- livebook:{"branch_parent_index":1} -->
92
+
box_count = length(boxes)
98
+
|> Enum.reduce_while([], fn pair, acc ->
99
+
new_acc = JunctionBoxes.group(acc, pair)
101
+
if MapSet.size(hd(new_acc)) == box_count do
116
+
<!-- livebook:{"offset":2222,"stamp":{"token":"XCP.sY5bJOGXuU_ZotobOVN6-zcg0TDJGEWVaJESs_-qDHzeEplFoStZ8_c6TRpS5eVpqnzoaj-gsLs_drksuYvMCDcM3T7_M6p9s09DRSKgMU2YB_Ogb0CB6_gAZamF2zEIUw","version":2}} -->