Advent-of-Code/2018/day01.py

36 lines
784 B
Python
Raw Normal View History

2019-11-14 16:23:47 +01:00
""" https://adventofcode.com/2018/day/1 """
def readFile():
import os.path as p
dName = p.dirname(__file__)
fName = p.basename(__file__).split(".")[0]
l = []
with open(p.join(dName, "input", f"{fName}.txt"), "r") as f:
line = f.readline()
while line:
l.append(int(line))
line = f.readline()
return l
def part1(vals : list):
sum = 0
for val in vals:
sum += val
return sum
def part2(vals : list):
sum = 0
s = {sum}
while True:
for val in vals:
sum += val
if sum in s:
return sum
s.add(sum)
return sum
if __name__ == "__main__":
vals = readFile()
print(f"Part 1: {part1(vals)}")
print(f"Part 2: {part2(vals)}")