summaryrefslogtreecommitdiffstats
path: root/solutions/py/d04.py
blob: c9bcbf314d4ba3a0197123ae704c6c0e030545ec (plain) (blame)
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
from collections import Counter

def isIncreasing(num):
    s = list(str(num))
    n = int(s[0])
    n_i = 0
    for sp in s[1:]:
        n_i += 1
        if int(sp) < n:
            for i in range(n_i, 6):
                s[i] = str(n)
            return False, int("".join(s))
        n = int(sp)
    return (True,)

def pt1(input):
    def containsDouble(num):
        s = str(num)
        amounts = [0 for _ in range(10)]
        for c in s:
            amounts[int(c)] += 1
        c = Counter(amounts)
        return c[0] + c[1] < 10

    amount = 0
    n = 357253
    while n < 892942 + 1:
        inc = isIncreasing(n)
        if inc[0] == True:
            if containsDouble(n):
                amount += 1
            n += 1
        else:
            n = inc[1]
    return amount

def pt2(input):
    def containsDouble(num):
        s = str(num)
        amounts = [0 for _ in range(10)]
        for c in s:
            amounts[int(c)] += 1
        c = Counter(amounts)
        if c[0] + c[1] < 10:
            return c[2] >= 1

    amount = 0
    n = 357253
    while n < 892942 + 1:
        inc = isIncreasing(n)
        if inc[0] == True:
            if containsDouble(n):
                amount += 1
            n += 1
        else:
            n = inc[1]
    return amount

if __name__ == "__main__":
    import cProfile

    cProfile.run("pt1([])")
    cProfile.run("pt2([])")
    print(pt1([]))
    print(pt2([]))