From f57a4810e22d446970c19286f8ebae4c9afa24ba Mon Sep 17 00:00:00 2001
From: Tomasz Kramkowski <tk@the-tk.com>
Date: Sat, 4 Dec 2021 14:54:10 +0000
Subject: day 4: get rid of won set
---
4.py | 4 +---
1 file changed, 1 insertion(+), 3 deletions(-)
diff --git a/4.py b/4.py
index 5e60e52..ed2f13c 100644
--- a/4.py
+++ b/4.py
@@ -34,17 +34,15 @@ class Board:
return Board([[int(n) for n in l.split()] for l in s.split('\n')])
def solve(nums: list[int], boards: list[Board]) -> tuple[int, int]:
- won: set[int] = set()
wins: list[int] = []
num: int
for num in nums:
i: int
board: Board
for i, board in enumerate(boards):
- if i in won: continue
+ if board.has_bingo: continue
board.call(num)
if board.has_bingo:
- won.add(i)
wins.append(num * board.unmarked_sum)
return wins[0], wins[-1]
--
cgit v1.2.3-70-g09d2