day25.py

# -*- coding: utf-8 -*-

import os
from time import perf_counter


def profiler(method):
    def profiler_method(*arg, **kw):
        t = perf_counter()
        ret = method(*arg, **kw)
        print(f'{method.__name__} method took : {perf_counter() - t:.4f} sec')
        return ret

    return profiler_method


# Part 1:
def part1(keys, locks) -> str | int:
    return sum(sum(all([k+l <= 5 for k, l in zip(key, lock)]) for lock in locks) for key in keys)


def get_input():
    with open(os.path.dirname(os.path.realpath(__file__)) + '/input', 'r', encoding='utf-8') as f:
        content = [s.strip() for s in f.read().rstrip().split('\n\n')]
    keys = []
    locks = []
    for item in content:
        rows = item.split('\n')
        top = rows.pop(0)
        rows.pop(-1)
        nums = [sum(['#' in i for i in l]) for l in zip(*rows)]
        if '.' in top and '#' not in top:
            keys.append(nums)
        else:
            locks.append(nums)
    return keys, locks


@profiler
def solve():
    print(f'Part 1: {part1(*get_input())}')


if __name__ == "__main__":
    solve()