day4.py

# vi: set shiftwidth=4 tabstop=4 expandtab:
import datetime
import os
import collections


top_dir = os.path.dirname(os.path.abspath(__file__)) + "/../../"


def is_valid(n):
    s = str(n)
    return (
        len(s) == 6
        and len(set(s)) <= 5
        and all(first <= second for first, second in zip(s, s[1:]))
    )


def is_valid2(n):
    s = str(n)
    return (
        len(s) == 6
        and 2 in collections.Counter(s).values()
        and all(first <= second for first, second in zip(s, s[1:]))
    )


def get_passwords(valid_range, func):
    mini, maxi = valid_range
    for n in range(mini, maxi):
        if func(n):
            yield n


def run_tests():
    assert is_valid(111111)
    assert not is_valid(223450)
    assert not is_valid(123789)
    assert is_valid2(112233)
    assert not is_valid2(123444)
    assert is_valid2(111122)


def get_solutions():
    input_range = 172930, 683082
    print(len(list(get_passwords(input_range, is_valid))) == 1675)
    print(len(list(get_passwords(input_range, is_valid2))) == 1142)


if __name__ == "__main__":
    begin = datetime.datetime.now()
    run_tests()
    get_solutions()
    end = datetime.datetime.now()
    print(end - begin)