Update day01.py

2020/Python/day01.py

@@ -1,12 +1,24 @@
-def solve_p2():
+def solve_p1():
     numbers = []
     with open("input.txt") as f:
         for line in f:
             num = int(line)
+            opp = 2020-num
+            if opp in numbers:
+                print(num*opp)
             numbers.append(num)
+
+
+def solve_p2():
+    numbers = []
+    with open("input.txt") as f:
+        for line in f:
+            numbers.append(int(line))
     combos = [((a, b, c), a*b*c) if a+b+c == 2020 else None for (a, b, c) in itertools.combinations(numbers, 3)]
     for combo in combos:
         if combo is not None:
             print(combo)
 
+            
+solve_p1():
 solve_p2()