2021 day10 part 2

This commit is contained in:
Adam Millerchip 2022-11-29 23:43:27 +09:00
parent 0a27136e95
commit 53fe13031e

View file

@ -7,11 +7,11 @@ defmodule Day10 do
|> Enum.sum() |> Enum.sum()
end end
def parse_line(_, {:corrupted, char}), do: {:corrupted, char} defp parse_line(_, {:corrupted, char}), do: {:corrupted, char}
def parse_line("", _), do: :incomplete defp parse_line("", stack), do: {:incomplete, stack}
def parse_line(<<char::utf8, rest::binary>>, []), do: parse_line(rest, [char]) defp parse_line(<<char::utf8, rest::binary>>, []), do: parse_line(rest, [char])
def parse_line(<<char::utf8, next::binary>>, [prev | popped] = stack) do defp parse_line(<<char::utf8, next::binary>>, [prev | popped] = stack) do
stack = stack =
case char do case char do
char when char in '([{<' -> [char | stack] char when char in '([{<' -> [char | stack]
@ -30,10 +30,28 @@ defmodule Day10 do
defp score(?}), do: 1197 defp score(?}), do: 1197
defp score(?>), do: 25137 defp score(?>), do: 25137
def part2(_input) do def part2(input) do
:ok scores =
input
|> Enum.map(&parse_line(&1, []))
|> Enum.filter(&match?({:incomplete, _}, &1))
|> Enum.map(fn {_, chars} -> score2(chars) end)
|> Enum.sort()
Enum.at(scores, trunc(length(scores) / 2))
end end
defp score2(chars) do
for char <- chars, reduce: 0 do
score -> score * 5 + points(char)
end
end
defp points(?(), do: 1
defp points(?[), do: 2
defp points(?{), do: 3
defp points(?<), do: 4
def input do def input do
with [input_filename] <- System.argv(), with [input_filename] <- System.argv(),
{:ok, input} <- File.read(input_filename) do {:ok, input} <- File.read(input_filename) do
@ -74,4 +92,4 @@ defmodule Day10 do
end end
end end
# Day10.run() Day10.run()