import matplotlib.pyplot as plt
with open("./out/chall.sol", "rt") as f:
public_inputs = f.readlines()[18][44:271].split("), uint(")
public_inputs = [int(public_inputs[i]) for i in range(len(public_inputs))]
# Traverse in reverse order using the origin as the starting point
maze = [public_inputs[i:i+5] for i in range(21, -1, -5)]
fig, ax = plt.subplots()
ax.imshow(maze, cmap="binary")