import sys output_two = 0 output_twelve = 0 file = '3.input' if len(sys.argv) <= 1 else sys.argv[1] for line in open(file).readlines(): line = line.strip() for n in [2, 12]: result = [-1] * n for it in range(n): right = list(enumerate(line[:-(n-1)+it or len(line)])) allowed = [(i, d) for i, d in right if i > max(result)] pick = max(allowed, key=lambda a: a[1]) result[it] = pick[0] if n == 2: output_two += int("".join(line[i] for i in result)) elif n == 12: output_twelve += int("".join(line[i] for i in result)) print(f'p1: {output_two}') print(f'p2: {output_twelve}')