AoC day 20

code.py

from __future__ import print_function

from itertools import chain, count
from math import sqrt

TARGET = 36000000

factors = lambda i: set(chain.from_iterable((x, i // x) for x in filter(lambda x: i % x == 0, range(1, int(sqrt(i)) + 1))))

part_one = False
part_two = False

for i in count(start=1):
    f = factors(i)
    if not part_one and sum(f) * 10 >= TARGET:
        print("Part One:", i)
        part_one = True
    if not part_two and sum(filter(lambda x: x * 50 >= i, f)) * 11 >= TARGET:
        print("Part Two:", i)
        part_two = True
    if part_one and part_two:
        break