You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.1 KiB
45 lines
1.1 KiB
3 years ago
|
import numpy as np
|
||
|
|
|
||
|
|
|
||
|
|
class Board:
|
||
|
|
|
||
|
|
def __init__(self, numbers):
|
||
|
|
self.values = np.reshape(numbers, (5, 5))
|
||
|
|
self.marked = np.zeros((5, 5))
|
||
|
|
|
||
|
|
def markNumber(self, num):
|
||
|
|
self.marked[self.values == num] = 1
|
||
|
|
|
||
|
|
def hasWon(self):
|
||
|
|
byColumn = np.max(np.min(self.marked, axis=0))
|
||
|
|
byRow = np.max(np.min(self.marked, axis=1))
|
||
|
|
return byRow or byColumn
|
||
|
|
|
||
|
|
def score(self):
|
||
|
|
return self.values[self.marked == 0].sum()
|
||
|
|
|
||
|
|
def __str__(self):
|
||
|
|
return str(self.marked)
|
||
|
|
|
||
|
|
|
||
|
|
with open("input", "r") as file:
|
||
|
|
numbersToCall = [int(strNum) for strNum in file.readline().split(",")]
|
||
|
|
allBoardNumbers = [int(strNum) for strNum in file.read().split()]
|
||
|
|
|
||
|
|
boards = [Board(allBoardNumbers[i:i + 25]) for i in range(0, len(allBoardNumbers), 25)]
|
||
|
|
|
||
|
|
startLen = len(boards)
|
||
|
|
for number in numbersToCall:
|
||
|
|
for board in boards[:]:
|
||
|
|
board.markNumber(number)
|
||
|
|
if board.hasWon():
|
||
|
|
if len(boards) == startLen:
|
||
|
|
finalScore = board.score() * number
|
||
|
|
print(f"First score: {finalScore}")
|
||
|
|
if len(boards) == 1:
|
||
|
|
finalScore = board.score() * number
|
||
|
|
print(f"Last score: {finalScore}")
|
||
|
|
boards.remove(board)
|
||
|
|
if len(boards) == 0:
|
||
|
|
break
|