summaryrefslogtreecommitdiff
path: root/day1/day1.py
diff options
context:
space:
mode:
Diffstat (limited to 'day1/day1.py')
-rw-r--r--day1/day1.py18
1 files changed, 18 insertions, 0 deletions
diff --git a/day1/day1.py b/day1/day1.py
new file mode 100644
index 0000000..b0623f9
--- /dev/null
+++ b/day1/day1.py
@@ -0,0 +1,18 @@
+from functools import reduce
+from itertools import combinations
+
+
+def mult(iterable):
+ return reduce(lambda x, y: x * y, iterable)
+
+
+def find_x_nums(data, sum_value, num):
+ for vals in combinations(data, num):
+ if sum(vals) == sum_value:
+ return vals
+
+
+data = [int(str.strip(x)) for x in open("input")]
+
+print(mult(find_x_nums(data, 2020, 2)))
+print(mult(find_x_nums(data, 2020, 3)))