29 lines
852 B
Python
Executable File
29 lines
852 B
Python
Executable File
#!/usr/bin/env python3
|
|
from useful import *
|
|
|
|
def win(b, d):
|
|
#print(b, d)
|
|
hit = np.vectorize(d.get)(b)
|
|
#print(hit)
|
|
size = b.shape[0]
|
|
return any(correct == size for correct in (*hit.sum(axis=0), *hit.sum(axis=1)))
|
|
|
|
def score(b, d, last):
|
|
return last * sum(x for x in b.reshape(np.prod(b.shape)) if not d[x])
|
|
|
|
with open(0) as f:
|
|
|
|
header, footer = f.read().split('\n\n', maxsplit=1)
|
|
pool = [int(x) for x in header.strip().split(',')]
|
|
boards = [np.asarray([int(n) for n in board.strip().split()]).reshape((5,5))
|
|
for board in footer.split('\n\n')]
|
|
#print(boards)
|
|
drawn = {n: False for n in pool} # to vectorize
|
|
for n in pool:
|
|
#print('%d...' % n)
|
|
drawn[n] = True
|
|
for b in boards:
|
|
if win(b, drawn):
|
|
print(score(b, drawn, n))
|
|
exit()
|