From 86eea5204564fb161faa96fcdd9580d7bd6e0d47 Mon Sep 17 00:00:00 2001 From: Tomasz Kramkowski Date: Sat, 2 Dec 2023 11:22:43 +0000 Subject: use operator --- 2.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/2.py b/2.py index 778208e..b5599f5 100644 --- a/2.py +++ b/2.py @@ -2,6 +2,7 @@ from sys import stdin from dataclasses import dataclass from functools import reduce from typing import TypeAlias +from operator import mul CubeSet: TypeAlias = dict[str, int] @@ -28,5 +29,5 @@ limits = {"red": 12, "green": 13, "blue": 14} # fmt: off print(sum(g.id for g in games if all(s[k] <= limits[k] for s in g.subsets for k in limits.keys() if k in s))) -print(sum(reduce(int.__mul__, (max(s[k] for s in g.subsets if k in s) for k in limits.keys())) for g in games)) +print(sum(reduce(mul, (max(s[k] for s in g.subsets if k in s) for k in limits.keys())) for g in games)) # fmt: on -- cgit v1.2.3-54-g00ecf