diff options
| -rw-r--r-- | 4.py | 13 |
1 files changed, 8 insertions, 5 deletions
@@ -34,16 +34,19 @@ 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]: + indices: set[int] = { i for i in range(len(boards)) } wins: list[int] = [] num: int for num in nums: i: int board: Board - for i, board in enumerate(boards): - if board.has_bingo: continue - board.call(num) - if board.has_bingo: - wins.append(num * board.unmarked_sum) + winners: set[int] = set() + for i in indices: + boards[i].call(num) + if boards[i].has_bingo: + wins.append(num * boards[i].unmarked_sum) + winners.add(i) + indices.difference_update(winners) return wins[0], wins[-1] nums: str | list[int] |