import aoclib import re if 'TEST' not in globals(): TEST = False my_input = aoclib.getInputLineAsArray(day=2, test=TEST) splitter = re.compile(r'([0-9]+)-([0-9]+) ([a-z]): (.*)') valid_count = 0 for line in my_input: match = re.match(splitter, line) (min_count, max_count, character, password) = match.group(1, 2, 3, 4) if (password[int(min_count) - 1] == character or password[int(max_count) - 1] == character) \ and password[int(min_count) - 1] != password[int(max_count) - 1]: valid_count = valid_count + 1 aoclib.printSolution(2, 1, valid_count, TEST)