Commit 4ea875f

Viliam PucikViliam Pucik

Viliam Pucik

authored and

Viliam Pucik

committed

2023-02

1 parent eb412b0 commit 4ea875fCopy full SHA for 4ea875f

File tree

+17

-0

lines changed

+17

-0

lines changed

Lines changed: 17 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,17 @@`
	`1`	`+#!/usr/bin/env python`
	`2`	`+import re`
	`3`	`+from math import prod`
	`4`	`+`
	`5`	`+s1, s2, balls, r = 0, 0, {"red": 12, "green": 13, "blue": 14}, re.compile(r"(\d+) (\w+)")`
	`6`	`+`
	`7`	`+for i, line in enumerate(open(0).read().splitlines(), start=1):`
	`8`	`+ maxes = {"red": 0, "green": 0, "blue": 0}`
	`9`	`+`
	`10`	`+ for count, color in r.findall(line):`
	`11`	`+ maxes[color] = max(maxes[color], int(count))`
	`12`	`+`
	`13`	`+ s1 += i * all(balls[color] >= count for color, count in maxes.items())`
	`14`	`+ s2 += prod(maxes.values())`
	`15`	`+`
	`16`	`+print(s1)`
	`17`	`+print(s2)`

Comments

(0)