[zerojudge] e998: S形矩陣

from sys import *

A = list(map(str, range(1, 10001)))
# 一次建好型態為 string 的表
opt = '' # 最後一次印
for s in stdin:
    n = int(s)
    for i in range(n):
        if(i%2==0): opt += ' '.join(A[i*n:i*n+n]) + ' \n'
        else: opt += ' '.join(A[i*n:i*n+n][::-1]) + ' \n'
        # i 為奇數時取出值，先反轉再 join
        # i 為偶數時，直接 join
    opt += '\n'
stdout.write(opt)