2023 Day 4 part 2

This commit is contained in:
Adam Millerchip 2023-12-04 20:11:26 +09:00
parent 1e74b21dc5
commit dce598de72

33
2023/day4.exs Normal file → Executable file
View file

@ -1,8 +1,9 @@
#!/usr/bin/env elixir
defmodule Day4 do defmodule Day4 do
def part1(input) do def part1(input) do
input input
|> Enum.map(fn {_id, winning, have} -> |> Enum.map(fn {_id, num_winners} ->
case MapSet.intersection(winning, have) |> MapSet.size() do case num_winners do
0 -> 0 0 -> 0
num_winners -> 2 ** (num_winners - 1) num_winners -> 2 ** (num_winners - 1)
end end
@ -10,8 +11,26 @@ defmodule Day4 do
|> Enum.sum() |> Enum.sum()
end end
def part2(_input) do def part2(input) do
:ok input
|> Enum.sort_by(fn {id, _num_winners} -> id end)
|> Enum.map(fn {_id, num_winners} -> {_num_copies = 1, num_winners} end)
|> copy()
|> Enum.map(fn {num_copies, _num_winners} -> num_copies end)
|> Enum.sum()
end
def copy([]), do: []
def copy([{above_num_copies, num_winners} = current | rest]) do
{to_copy, left_alone} = Enum.split(rest, num_winners)
copied =
Enum.map(to_copy, fn {below_num_copies, num_winners} ->
{above_num_copies + below_num_copies, num_winners}
end)
[current | copy(copied ++ left_alone)]
end end
def input do def input do
@ -27,7 +46,9 @@ defmodule Day4 do
for cards <- [winning, have], for cards <- [winning, have],
do: cards |> String.split(" ", trim: true) |> MapSet.new(&String.to_integer/1) do: cards |> String.split(" ", trim: true) |> MapSet.new(&String.to_integer/1)
{String.to_integer(id), winning, have} num_winners = MapSet.intersection(winning, have) |> MapSet.size()
{String.to_integer(id), num_winners}
end) end)
else else
_ -> :error _ -> :error
@ -65,4 +86,4 @@ defmodule Day4 do
end end
end end
# Day4.run() Day4.run()