Codeup 1504 - 지그재그 배열 2

http://codeup.kr/problem.php?id=1504

 

Codeup 1504 - 지그재그 배열 2

 

n = int(input())
matrix = [[0]*n for i in range(n)]
cnt = 0
​
for i in range(0, n):
    if i % 2:
        for j in range(n-1, -1, -1):
            cnt += 1
            matrix[j][i] = cnt
    else:
        for j in range(0, n):
            cnt += 1
            matrix[j][i] = cnt
​
for i in range(0, n):
    for j in range(0, n):
        print(matrix[i][j], end=' ')
    print()