summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--4.py4
1 files changed, 1 insertions, 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]