adventofcode/2.py at main · chrispoch/adventofcode · GitHub

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
import os


#initial
with open("2.txt") as file:
    lines = file.readlines()
    total = 0
    valid = 0
    invalid = 0

    for line in lines:
        total = total + 1

        sep = line.find("-")
        space = line.find(" ")
        min = int(line[0:sep])
        max = int(line[sep + 1:space])
        letter = line[space + 1]
        password = line[space + 4:]

        found = password.count(letter)
        if found >= min and found <= max:
            valid = valid + 1
        else:
            invalid = invalid + 1
    print(total, valid, invalid)

#part2
    total = 0
    valid = 0
    invalid = 0

    for line in lines:
        total = total + 1

        sep = line.find("-")
        space = line.find(" ")
        pos1 = int(line[0:sep]) - 1
        pos2 = int(line[sep + 1:space]) - 1
        letter = line[space + 1]
        password = line[space + 4:]
        #print (line, str(pos1), str(sep), str(pos2), str(space), letter, password)

        if (password[pos1] == letter or password[pos2] == letter) and (password[pos1] != password[pos2]):
            valid = valid + 1
        else:
            invalid = invalid + 1
    print(total, valid, invalid)