diff --git a/2020/day1.py b/2020/day1.py new file mode 100644 index 0000000..40378f0 --- /dev/null +++ b/2020/day1.py @@ -0,0 +1,20 @@ +import itertools + + +def main(): + with open("day1.txt", "rt") as f: + lines = [line.rstrip() for line in f.readlines()] + values = list(map(lambda l: int(l), lines)) + + # Part 1 + for a, b in itertools.combinations(values, 2): + if a + b == 2020: + print(a * b) + # Part 2 + for a, b, c in itertools.combinations(values, 3): + if a + b + c == 2020: + print(a * b * c) + + +if __name__ == "__main__": + main() diff --git a/2020/skel.py.txt b/2020/skel.py.txt new file mode 100644 index 0000000..732477d --- /dev/null +++ b/2020/skel.py.txt @@ -0,0 +1,7 @@ +def main(): + with open("", "rt") as f: + lines = [line.rstrip() for line in f.readlines()] + + +if __name__ == "__main__": + main()