Codeup 1470 - [기초-배열연습] 2차원 배열 지그재그 채우기 2-3

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

 

Codeup 1470 - [기초-배열연습] 2차원 배열 지그재그 채우기 2-3

 

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[i][j] = cnt
    else:
        for j in range(0, n):
            cnt += 1
            matrix[i][j] = cnt
​
for i in range(0, n):
    for j in range(0, n):
        print(matrix[j][i], end=' ')
    print()