65 lines
1.3 KiB
Elixir
65 lines
1.3 KiB
Elixir
defmodule Day1 do
|
|
def calculate_frequency(shifts) do
|
|
shifts
|
|
|> Enum.reduce(0, &+/2)
|
|
end
|
|
|
|
def first_repetition(shifts, index \\ 0, sum \\ 0, seen \\ [0]) do
|
|
sum = sum + Enum.at(shifts, index)
|
|
next_index = rem(index + 1, Enum.count(shifts))
|
|
|
|
cond do
|
|
Enum.member?(seen, sum) ->
|
|
sum
|
|
|
|
true ->
|
|
first_repetition(shifts, next_index, sum, [sum | seen])
|
|
end
|
|
end
|
|
|
|
def first_repetition_stream(shifts) do
|
|
shifts
|
|
|> Stream.cycle()
|
|
|> Enum.reduce_while({0, [0]}, fn shift, {sum, seen} ->
|
|
sum = sum + shift
|
|
|
|
cond do
|
|
Enum.member?(seen, sum) -> {:halt, sum}
|
|
true -> {:cont, {sum, [sum | seen]}}
|
|
end
|
|
end)
|
|
end
|
|
|
|
def first_repetition_mapset(shifts) do
|
|
shifts
|
|
|> Stream.cycle()
|
|
|> Enum.reduce_while({0, MapSet.new([0])}, fn shift, {sum, seen} ->
|
|
sum = sum + shift
|
|
|
|
cond do
|
|
sum in seen -> {:halt, sum}
|
|
true -> {:cont, {sum, MapSet.put(seen, sum)}}
|
|
end
|
|
end)
|
|
end
|
|
|
|
def parse_input() do
|
|
File.read!("lib/day1/input.txt")
|
|
|> String.split("\n")
|
|
|> Enum.map(fn s ->
|
|
{num, _} = Integer.parse(s)
|
|
num
|
|
end)
|
|
end
|
|
|
|
def part1() do
|
|
parse_input()
|
|
|> calculate_frequency()
|
|
end
|
|
|
|
def part2() do
|
|
parse_input()
|
|
|> first_repetition_mapset()
|
|
end
|
|
end
|