66 lines
1.4 KiB
Python
66 lines
1.4 KiB
Python
with open('day13-input', 'r') as file:
|
|
data = [l.strip('\n') for l in file]
|
|
import numpy as np
|
|
|
|
grid = np.zeros((len(data[0]), len(data)), dtype=np.int64)
|
|
minecarts = []
|
|
|
|
codes = {
|
|
' ': 0,
|
|
'-': 1,
|
|
'|': 2,
|
|
'+': 3,
|
|
'/': 4,
|
|
'\\': 5,
|
|
}
|
|
carts = {
|
|
'>': (0, 1),
|
|
'v': (1, 2),
|
|
'<': (2, 1),
|
|
'^': (3, 2),
|
|
}
|
|
|
|
directions = np.array([[1,0], [0,1], [-1,0], [0,-1]])
|
|
turns = [-1, 0, 1]
|
|
|
|
for y, line in enumerate(data):
|
|
for x, char in enumerate(line):
|
|
if char in codes:
|
|
grid[x,y] = codes[char]
|
|
else:
|
|
grid[x,y] = carts[char][1]
|
|
minecarts.append([x, y, carts[char][0], 0])
|
|
|
|
while len(minecarts) > 1:
|
|
minecarts = sorted(minecarts, key=lambda c: (c[1], c[0]))
|
|
i = 0
|
|
while i < len(minecarts):
|
|
mc = minecarts[i]
|
|
mc[:2] = list(mc[:2]+directions[mc[2]])
|
|
g = grid[tuple(mc[:2])]
|
|
if g in (1, 2):
|
|
pass
|
|
elif g == 4:
|
|
mc[2] = (3, 2, 1, 0)[mc[2]]
|
|
elif g == 5:
|
|
mc[2] = (1, 0, 3, 2)[mc[2]]
|
|
elif g == 3:
|
|
mc[2] = (mc[2]+turns[mc[3]]) % 4
|
|
mc[3] = (mc[3]+1) % 3
|
|
for j, mc2 in enumerate(minecarts):
|
|
if mc2 == mc:
|
|
continue
|
|
if np.array_equal(mc[:2], mc2[:2]):
|
|
print('Crash at', mc[:2])
|
|
if j > i:
|
|
minecarts.pop(j)
|
|
minecarts.pop(i)
|
|
i -= 1
|
|
else:
|
|
minecarts.pop(i)
|
|
minecarts.pop(j)
|
|
i -= 2
|
|
break
|
|
i += 1
|
|
print('Last cart', minecarts[0])
|