Spaces:
Running
Running
File size: 1,529 Bytes
a4da721 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 |
def load_data(file):
with open(file) as f:
data = f.readlines()
# Add newline to make it easier to parse
data = data + ["\n"]
keys = []
locks = []
grid = []
for idx, line in enumerate(data):
line = line.strip("\n")
if len(line) == 0:
if grid[0][0] == "#":
locks.append(grid)
else:
keys.append(grid)
grid = []
continue
grid.append(line)
return keys, locks
def get_lock_profile(lock):
M = len(lock)
N = len(lock[0])
profile = []
for j in range(N):
for i in range(M):
if lock[i][j] == ".":
profile.append(i-1)
break
return profile
def get_key_profile(key):
M = len(key)
N = len(key[0])
profile = []
for j in range(N):
# for j in range(N-1, -1, -1):
for i in range(M):
if key[i][j] == "#":
profile.append(6-i)
break
return profile
def check_fit(key, lock):
for k, l in zip(key, lock):
if k + l > 5:
return False
return True
# file = "test.txt"
file = "input.txt"
keys, locks = load_data(file)
key_profiles = []
for key in keys:
key_profiles.append(get_key_profile(key))
lock_profiles = []
for lock in locks:
lock_profiles.append(get_lock_profile(lock))
fits = []
for key in key_profiles:
for lock in lock_profiles:
fits.append(check_fit(key, lock))
print(sum(fits))
|