···
1
+
<!-- livebook:{"persist_outputs":true} -->
7
+
{:kino_aoc, git: "https://github.com/ljgago/kino_aoc"}
11
+
<!-- livebook:{"output":true} -->
19
+
<!-- livebook:{"attrs":{"day":"11","session_secret":"ADVENT_OF_CODE_SESSION","variable":"puzzle_input","year":"2022"},"kind":"Elixir.KinoAOC.HelperCell","livebook_object":"smart_cell"} -->
22
+
{:ok, puzzle_input} =
23
+
KinoAOC.download_puzzle("2022", "11", System.fetch_env!("LB_ADVENT_OF_CODE_SESSION"))
26
+
<!-- livebook:{"output":true} -->
30
+
"Monkey 0:\n Starting items: 93, 98\n Operation: new = old * 17\n Test: divisible by 19\n If true: throw to monkey 5\n If false: throw to monkey 3\n\nMonkey 1:\n Starting items: 95, 72, 98, 82, 86\n Operation: new = old + 5\n Test: divisible by 13\n If true: throw to monkey 7\n If false: throw to monkey 6\n\nMonkey 2:\n Starting items: 85, 62, 82, 86, 70, 65, 83, 76\n Operation: new = old + 8\n Test: divisible by 5\n If true: throw to monkey 3\n If false: throw to monkey 0\n\nMonkey 3:\n Starting items: 86, 70, 71, 56\n Operation: new = old + 1\n Test: divisible by 7\n If true: throw to monkey 4\n If false: throw to monkey 5\n\nMonkey 4:\n Starting items: 77, 71, 86, 52, 81, 67\n Operation: new = old + 4\n Test: divisible by 17\n If true: throw to monkey 1\n If false: throw to monkey 6\n\nMonkey 5:\n Starting items: 89, 87, 60, 78, 54, 77, 98\n Operation: new = old * 7\n Test: divisible by 2\n If true: throw to monkey 1\n If false: throw to monkey 4\n\nMonkey 6:\n Starting items: 69, 65, 63\n Operation: new = old + 6\n Test: divisible by 3\n If true: throw to monkey 7\n If false: throw to monkey 2\n\nMonkey 7:\n Starting items: 89\n Operation: new = old * old\n Test: divisible by 11\n If true: throw to monkey 0\n If false: throw to monkey 2\n"}
35
+
defstruct items: [],
36
+
operation: &Function.identity/1,
45
+
"Starting items: " <> items,
46
+
"Operation: new = old " <> operation,
47
+
"Test: divisible by " <> test,
48
+
"If true: throw to monkey " <> if_true,
49
+
"If false: throw to monkey " <> if_false
52
+
|> String.split("\n", trim: true)
53
+
|> Enum.map(&String.trim/1)
56
+
items: parse_items(items),
57
+
operation: parse_operation(operation),
58
+
test: String.to_integer(test),
59
+
true: String.to_integer(if_true),
60
+
false: String.to_integer(if_false)
64
+
def run(%__MODULE__{} = monkey, calming, lcm \\ 1) do
65
+
{true_val, false_val} =
67
+
|> Enum.map(fn item ->
69
+
|> then(monkey.operation)
73
+
|> Enum.split_with(fn item ->
74
+
rem(item, monkey.test) == 0
77
+
{struct(monkey, passes: monkey.passes + length(monkey.items), items: []),
78
+
{monkey.true, true_val}, {monkey.false, false_val}}
81
+
def add_items(%__MODULE__{} = monkey, new_items) do
82
+
struct(monkey, items: monkey.items ++ new_items)
85
+
defp parse_items(items) do
87
+
|> String.split(", ")
88
+
|> Enum.map(&String.to_integer/1)
91
+
defp parse_operation("* old"),
92
+
do: fn old -> old * old end
94
+
defp parse_operation("* " <> num) do
95
+
num = String.to_integer(num)
97
+
fn old -> old * num end
100
+
defp parse_operation("+ " <> num) do
101
+
num = String.to_integer(num)
103
+
fn old -> old + num end
108
+
<!-- livebook:{"output":true} -->
111
+
{:module, Monkey, <<70, 79, 82, 49, 0, 0, 24, ...>>, {:parse_operation, 1}}
117
+
|> String.split("\n\n")
118
+
|> Enum.map(&Monkey.parse/1)
121
+
<!-- livebook:{"output":true} -->
127
+
operation: #Function<3.112681941/1 in Monkey.parse_operation/1>,
135
+
operation: #Function<1.112681941/1 in Monkey.parse_operation/1>,
143
+
operation: #Function<1.112681941/1 in Monkey.parse_operation/1>,
151
+
operation: #Function<1.112681941/1 in Monkey.parse_operation/1>,
159
+
operation: #Function<1.112681941/1 in Monkey.parse_operation/1>,
167
+
operation: #Function<3.112681941/1 in Monkey.parse_operation/1>,
175
+
operation: #Function<1.112681941/1 in Monkey.parse_operation/1>,
183
+
operation: #Function<2.112681941/1 in Monkey.parse_operation/1>,
193
+
defmodule MonkeyBusiness do
194
+
import Kernel, except: [round: 1]
196
+
defstruct monkeys: [], lcm: nil
198
+
def new(monkeys) do
199
+
lcm = Enum.reduce(monkeys, 1, &lcm(&1.test, &2))
207
+
def run(%__MODULE__{} = mb, rounds, calming) do
209
+
|> Enum.reduce(mb, fn r, mb ->
210
+
# IO.inspect(r, label: :round)
213
+
|> Map.get(:monkeys)
214
+
|> Enum.map(& &1.passes)
215
+
|> Enum.sort(:desc)
217
+
|> Enum.reduce(&*/2)
220
+
defp round(%__MODULE__{} = mb, calming) do
221
+
0..(length(mb.monkeys) - 1)
222
+
|> Enum.reduce(mb, fn idx, mb ->
223
+
{monkey, {a, a_val}, {b, b_val}} =
226
+
|> Monkey.run(calming, mb.lcm)
229
+
|> put_in([Access.key(:monkeys), Access.at(idx)], monkey)
230
+
|> update_in([Access.key(:monkeys), Access.at(a)], &Monkey.add_items(&1, a_val))
231
+
|> update_in([Access.key(:monkeys), Access.at(b)], &Monkey.add_items(&1, b_val))
235
+
defp lcm(a, b), do: div(a * b, Integer.gcd(a, b))
239
+
<!-- livebook:{"output":true} -->
242
+
warning: variable "r" is unused (if the variable is not meant to be used, prefix it with an underscore)
243
+
2022/day11.livemd#cell:wtmu5fny2ud5f5f6xet6khgysonm2y3n:17: MonkeyBusiness.run/3
247
+
<!-- livebook:{"output":true} -->
250
+
{:module, MonkeyBusiness, <<70, 79, 82, 49, 0, 0, 21, ...>>, {:lcm, 2}}
257
+
|> MonkeyBusiness.new()
258
+
|> MonkeyBusiness.run(20, 3)
261
+
<!-- livebook:{"output":true} -->
271
+
|> MonkeyBusiness.new()
272
+
|> MonkeyBusiness.run(10000, 1)
275
+
<!-- livebook:{"output":true} -->
hauleth.dev