25.py

locks, keys = [], []
temp = []


def get_group(group):
    return locks if set(group[0]) == set("#") else keys


with open("data/25.txt") as file:
    for line in file:
        line = line.strip()
        if line:
            temp.append(line)
        else:
            get_group(temp).append(temp)
            temp = []

if temp:
    get_group(temp).append(temp)


intlocks = [
    [
        [lock[row][col] for row in range(len(lock))].index(".") - 1
        for col in range(len(lock[0]))
    ]
    for lock in locks
]


intkeys = [
    [
        [key[row][col] for row in range(len(key) - 1, -1, -1)].index(".") - 1
        for col in range(len(key[0]))
    ]
    for key in keys
]


def isoverlap(lock, key, maxheight):
    for i in range(len(lock)):
        if lock[i] + key[i] >= maxheight:
            return False
    return True


print(
    sum(isoverlap(lock, key, len(locks[0]) - 1) for lock in intlocks for key in intkeys)
)