69 lines
2 KiB
Elixir
69 lines
2 KiB
Elixir
defmodule Day9Part2 do
|
|
def run do
|
|
input =
|
|
File.read!("input")
|
|
|> String.split("\n", trim: true)
|
|
|> Enum.map(&String.to_integer/1)
|
|
|
|
invalid_number = Day9Part1.find_invalid(input)
|
|
|
|
input
|
|
|> find_weakness(invalid_number)
|
|
|> IO.puts()
|
|
end
|
|
|
|
def find_weakness([current | rest], invalid_number) do
|
|
case find_group(rest, [current], current, invalid_number) do
|
|
{:ok, group} -> Enum.min(group) + Enum.max(group)
|
|
_ -> find_weakness(rest, invalid_number)
|
|
end
|
|
end
|
|
|
|
def find_group(_list, group, sum, target) when sum == target, do: {:ok, group}
|
|
def find_group(_list, _group, sum, target) when sum > target, do: :too_big
|
|
def find_group(_list, [], _sum, _target), do: :eol
|
|
|
|
def find_group([next | rest], group, sum, target) do
|
|
find_group(rest, [next | group], sum + next, target)
|
|
end
|
|
end
|
|
|
|
defmodule Day9Part1 do
|
|
def run do
|
|
File.read!("input")
|
|
|> String.split("\n", trim: true)
|
|
|> Enum.map(&String.to_integer/1)
|
|
|> find_invalid()
|
|
|> IO.puts()
|
|
end
|
|
|
|
def find_invalid([_ | next] = numbers) do
|
|
{preamble, [number | _]} = Enum.split(numbers, 25)
|
|
sorted_preamble = Enum.sort(preamble)
|
|
max = number - hd(sorted_preamble)
|
|
|
|
if sum_in_preamble?(sorted_preamble, max, number) do
|
|
find_invalid(next)
|
|
else
|
|
number
|
|
end
|
|
end
|
|
|
|
def sum_in_preamble?([], _max, _number), do: false
|
|
def sum_in_preamble?([preamble_head | _], max, _number) when preamble_head >= max, do: false
|
|
|
|
def sum_in_preamble?([preamble_head | preamble_rest], max, number) do
|
|
if forms_pair?(preamble_head, preamble_rest, max, number) do
|
|
true
|
|
else
|
|
sum_in_preamble?(preamble_rest, max, number)
|
|
end
|
|
end
|
|
|
|
def forms_pair?(_x, [y | _], max, _number) when y > max, do: false
|
|
def forms_pair?(_x, [], _max, _number), do: false
|
|
def forms_pair?(x, [y | _], _max, number) when x + y == number, do: true
|
|
def forms_pair?(x, [_y | rest], max, number), do: forms_pair?(x, rest, max, number)
|
|
end
|
|
|
|
Day9Part2.run()
|