I’m providing the solution for Python and JS, please leave on the comments if you found a better way.
I really got the solution inspired by the image on the problem statement above. The idea is to resolve one “layer” at a time.
For each layer I’m building a flat array with the elements assigned to the correct position as it’s easier to do this on 1 dimension. After I complete the array, I then proceed to recreate the matrix layer from it.
The problem was actually quite simple to solve, though it looks messy.
I like using variables to define the direction where I’m going while looping through the matrix layer, though some solutions for problems like this usually don’t use this directional variables, but multiple loops instead. anyway works.
Here is my solution as always for Python and JS