Day 1: add alternate part 2 solution

This commit is contained in:
Shadowfacts 2018-12-01 14:11:51 -05:00
parent 3db1e578f0
commit aedc078ad7
Signed by: shadowfacts
GPG Key ID: 94A5AB95422746E5
2 changed files with 21 additions and 1 deletions

View File

@ -16,6 +16,18 @@ defmodule Day1 do
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 parse_input() do
File.read!("lib/day1/input.txt")
|> String.split("\n")
@ -32,6 +44,6 @@ defmodule Day1 do
def part2() do
parse_input()
|> first_repetition()
|> first_repetition_stream()
end
end

View File

@ -16,4 +16,12 @@ defmodule Day1Test do
assert Day1.first_repetition([-6, 3, 8, 5, -6]) == 5
assert Day1.first_repetition([7, 7, -2, -7, -4]) == 14
end
test "first repetition stream" do
assert Day1.first_repetition_stream([1, -2, 3, 1]) == 2
assert Day1.first_repetition_stream([1, -1]) == 0
assert Day1.first_repetition_stream([3, 3, 4, -2, -4]) == 10
assert Day1.first_repetition_stream([-6, 3, 8, 5, -6]) == 5
assert Day1.first_repetition_stream([7, 7, -2, -7, -4]) == 14
end
end