diff options
Diffstat (limited to 'day1/day1.py')
-rw-r--r-- | day1/day1.py | 18 |
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))) |