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
|
#!/usr/bin/env python3
from re import findall
from functools import reduce
# filename = "in/day05.ref"
filename = "in/day05.pzl"
m = [ [i for i in line.rstrip('\n')[1::4] ] for line in open(filename) if line.find('[') != -1]
# print('m:', m)
mt = [[m[col_n][row_n] for col_n,_ in enumerate(m) if m[col_n][row_n]!=' '][::-1] for row_n,_ in enumerate(m[0])]
# print('mt:', mt)
a = [ [int(val) if i==0 else int(val)-1 for i,val in enumerate(findall('\d+', line))] for line in open(filename) if line[0] == 'm']
# print('a:', a)
b1 = reduce(lambda state, action: [row[:-action[0]] if row_n==action[1] else row+state[action[1]][-action[0]:][::-1] if row_n==action[2] else row for row_n, row in enumerate(state)], a, mt)
# print('b1:', b1)
res1 = ''.join([i[-1] for i in b1])
print('res1:', res1)
b2 = reduce(lambda state, action: [row[:-action[0]] if row_n==action[1] else row+state[action[1]][-action[0]:] if row_n==action[2] else row for row_n, row in enumerate(state)], a, mt)
# print('b2:', b2)
res2 = ''.join([i[-1] for i in b2])
print('res2:', res2)
|
