day5 oneline

1 files changed, 28 insertions, 0 deletions

diff --git a/2022/day5/oneline.py b/2022/day5/oneline.py
new file mode 100755
index 0000000..736492a
--- /dev/null
+++ b/2022/day5/oneline.py
@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+
+from re import findall
+from functools import reduce
+
+# filename = "ref.txt"
+filename = "pzl.txt"
+
+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)
+