diff --git a/2020/day2.py b/2020/day2.py new file mode 100644 index 0000000..c559930 --- /dev/null +++ b/2020/day2.py @@ -0,0 +1,27 @@ +def main(): + with open("day2.txt", "rt") as f: + lines = [line.rstrip() for line in f.readlines()] + # Part 1 + valid = 0 + for line in lines: + count, letter, password = line.split() + min_n, max_n = count.split("-") + letter = letter.rstrip(":") + if int(min_n) <= password.count(letter) <= int(max_n): + valid += 1 + print("Valids:", valid) + # Part 2 + valid = 0 + for line in lines: + count, letter, password = line.split() + ofs_a, ofs_b = count.split("-") + letter = letter.rstrip(":") + a = password[int(ofs_a) - 1] + b = password[int(ofs_b) - 1] + if (a == letter or b == letter) and not (a == b == letter): + valid += 1 + print("Valids:", valid) + + +if __name__ == "__main__": + main()