28 lines
827 B
Elixir
28 lines
827 B
Elixir
|
defmodule Part2 do
|
||
|
|
def mark_seen(change, {prev_freq, seen}) do
|
||
|
|
freq = prev_freq + change
|
||
|
|
|
||
|
|
marked_freq = {freq, MapSet.member?(seen, freq)}
|
||
|
|
acc = {freq, MapSet.put(seen, freq)}
|
||
|
|
|
||
|
|
{[marked_freq], acc}
|
||
|
|
end
|
||
|
|
|
||
|
|
def run do
|
||
|
|
File.stream!("input")
|
||
|
|
|> Stream.map(&String.trim/1)
|
||
|
|
# Trying to work out why can't cycle over the stream
|
||
|
|
# Can't reproduce with this
|
||
|
|
# Stream.cycle([1]) |> Stream.transform(0, fn one, prev -> {[prev + one], prev + one} end) |> Enum.find(fn x -> x == 25 end)
|
||
|
|
# So maybe it's because a combination of the accumulator in the stream and Enum.find
|
||
|
|
|> Stream.map(&String.to_integer/1)
|
||
|
|
|> Stream.cycle()
|
||
|
|
|> Stream.transform({0, MapSet.new([0])}, &mark_seen/2)
|
||
|
|
|> Enum.find(fn {_freq, seen} -> seen end)
|
||
|
|
|> elem(0)
|
||
|
|
|> IO.puts()
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
Part2.run()
|