aoc/2022
1
0
Fork 0
Code Releases Activity

day 5: we do a little bit of numpy

Browse source
This commit is contained in:
Sijmen 2022-12-05 17:57:08 +01:00
parent 612b61a1ac
commit 354dac8de7
Signed by: vijfhoek
GPG key ID: DAF7821E067D9C48
1 changed files with 18 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
day05.py
View file

@ -1,27 +1,26 @@
import fileinput
import sys import sys
from collections import defaultdict, deque import numpy
import re
from copy import deepcopy
input = sys.stdin.read().split("\n") inp_stacks, inp_moves = (x.split("\n") for x in sys.stdin.read().split("\n\n", 1))
rows = (line[1::4] for line in input[::-1] if "[" in line) rows = (line[1::4] for line in inp_stacks[::-1] if "[" in line)
stacks = [[c for c in x if c != ' '] for x in zip(*rows)] stacks = [numpy.array([ord(c) for c in x if c != ' '], dtype=numpy.int8) for x in zip(*rows)]
stacks2 = deepcopy(stacks) stacks1 = numpy.array(stacks)
stacks2 = numpy.ndarray.copy(stacks1)
RE = re.compile(r"move (\d+) from (\d+) to (\d+)") for i, line in enumerate(inp_moves):
for line in input: if i % 10000 == 0:
m = RE.match(line) print(i, len(inp_moves), end="\r")
if not m: if not line:
continue continue
n, f, t = int(m[1]), int(m[2]) - 1, int(m[3]) - 1 _, a, _, b, _, c = line.split()
n, f, t = int(a), int(b) - 1, int(c) - 1
stacks[t].extend(reversed(stacks[f][-n:])) stacks1[t] = numpy.concatenate((stacks1[t], stacks1[f][-n:][::-1]))
del stacks[f][-n:] stacks1[f] = stacks1[f][:-n]
stacks2[t].extend(stacks2[f][-n:]) stacks2[t] = numpy.concatenate((stacks2[t], stacks2[f][-n:]))
del stacks2[f][-n:] stacks2[f] = stacks2[f][:-n]
print("".join(s[-1] for s in stacks), "".join(s[-1] for s in stacks2)) print("".join(chr(s[-1]) for s in stacks1), "".join(chr(s[-1]) for s in stacks2))