This commit is contained in:
basicbonobo 2023-12-14 14:40:58 +01:00
parent 115d295589
commit 3e95018ebb
1 changed files with 13 additions and 1 deletions

View File

@ -8,8 +8,17 @@ def is_mirror(pic, axis):
c = a == b
return c.all()
def is_smudged(pic, axis):
a = pic[:axis]
b = pic[i:i*2][::-1]
c = a != b
#ic(c)
#ic(c.sum())
#assert False
return c.sum() == 1
ms = [np.array([[c == '#' for c in l] for l in p.strip().split()]) for p in pgphs(open(0).read())]
score = 0
score, score2 = 0, 0
for m in ms:
for tran, scoretran in ((lambda a: a, lambda n, h: n * 100),
(lambda a: a[::-1], lambda n, h: (h - n) * 100),
@ -20,4 +29,7 @@ for m in ms:
if is_mirror(tran(m), i):
#ic(tran, i, draw(m))
score += scoretran(i, height)
if is_smudged(tran(m), i):
score2 += scoretran(i, height)
print(score)
print(score2)