-
Notifications
You must be signed in to change notification settings - Fork 0
/
day1.py
44 lines (31 loc) · 1001 Bytes
/
day1.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# vi: set shiftwidth=4 tabstop=4 expandtab:
import datetime
import os
top_dir = os.path.dirname(os.path.abspath(__file__)) + "/../../"
def get_freq_from_file(file_path=top_dir + "resources/year2018_day1_input.txt"):
with open(file_path) as f:
return [int(l.strip()) for l in f]
def get_resulting_freq(freqs):
return sum(freqs)
def get_first_repetition(freqs):
freq = 0
seens = set([freq])
while True:
for f in freqs:
freq += f
if freq in seens:
return freq
seens.add(freq)
def run_tests():
assert get_resulting_freq([+1, -2, +3, +1]) == 3
assert get_first_repetition([+1, -2, +3, +1]) == 2
def get_solutions():
freqs = get_freq_from_file()
print(get_resulting_freq(freqs) == 587)
print(get_first_repetition(freqs) == 83130)
if __name__ == "__main__":
begin = datetime.datetime.now()
run_tests()
get_solutions()
end = datetime.datetime.now()
print(end - begin)