kot.pink
1import sys
2
3output_two = 0
4output_twelve = 0
5file = '3.input' if len(sys.argv) <= 1 else sys.argv[1]
6for line in open(file).readlines():
7 line = line.strip()
8 for n in [2, 12]:
9 result = [-1] * n
10 for it in range(n):
11 right = list(enumerate(line[:-(n-1)+it or len(line)]))
12 allowed = [(i, d) for i, d in right if i > max(result)]
13 pick = max(allowed, key=lambda a: a[1])
14 result[it] = pick[0]
15
16 if n == 2:
17 output_two += int("".join(line[i] for i in result))
18 elif n == 12:
19 output_twelve += int("".join(line[i] for i in result))
20
21print(f'p1: {output_two}')
22print(f'p2: {output_twelve}')