1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
|
from typing import NamedTuple
from utils import open_day
class Win(NamedTuple):
index: int | None
round: int
Board = list[list[int]]
def board_from_string(s: str) -> Board:
return [[numidxs[int(cell)] for cell in row.split()] for row in s.split('\n')]
def resolve(nums: list[int], boards: list[Board], index: int, round: int) -> int:
board: Board = boards[index]
count: int = sum(nums[cell] for row in board for cell in row if cell > round)
return count * nums[round]
nums: str | list[int]
boards: list[str] | list[Board]
nums, *boards = open_day(4).read().rstrip().split('\n\n')
nums = list(map(int, nums.split(',')))
numidxs = { n: i for i, n in enumerate(nums) }
boards = [board_from_string(board) for board in boards]
first: Win = Win(None, max(nums))
last: Win = Win(None, 0)
for i, board in enumerate(boards):
win: Win = Win(i, min(
min(max(row) for row in board),
min(max(col) for col in zip(*board))
))
if win.round < first.round: first = win
if win.round > last.round: last = win
assert(first.index is not None and last.index is not None)
print(resolve(nums, boards, first.index, first.round))
print(resolve(nums, boards, last.index, last.round))
|
