aboutsummaryrefslogtreecommitdiff
path: root/2015/17/puzzle-2.py
diff options
context:
space:
mode:
Diffstat (limited to '2015/17/puzzle-2.py')
-rwxr-xr-x2015/17/puzzle-2.py29
1 files changed, 22 insertions, 7 deletions
diff --git a/2015/17/puzzle-2.py b/2015/17/puzzle-2.py
index 43f8922..50a70f9 100755
--- a/2015/17/puzzle-2.py
+++ b/2015/17/puzzle-2.py
@@ -7,13 +7,28 @@ def main() -> None:
with open("input", "r", encoding="utf-8") as f:
nums = list(map(int, f.readlines()))
- combs: list[tuple[int, ...]] = []
- for n in range(len(nums) + 1):
- combs += list(itertools.combinations(nums, n))
-
- combs = list(filter(lambda x: sum(x) == 150, combs))
- conts = min(map(len, combs))
- print(len(list(filter(lambda x: len(x) == conts, combs))))
+ print(
+ len(
+ list(
+ filter(
+ lambda x: len(x) == min(map(len, combs)),
+ (
+ combs := list(
+ filter(
+ lambda x: sum(x) == 150,
+ itertools.chain(
+ *[
+ list(itertools.combinations(nums, n))
+ for n in range(len(nums) + 1)
+ ]
+ ),
+ )
+ )
+ ),
+ )
+ )
+ )
+ )
if __name__ == "__main__":