Fastest Python solution I could figure out - Zigzag Conversion