[코드트리] 2차원 배열과 패턴 > 지그재그로 숫자 채우기(★)

n * m크기의 직사각형에 숫자를 0부터 순서대로 1씩 증가시키며 왼쪽 위에서부터 시작하여 다음과 같이 지그재그 모양으로 숫자들을 쭉 채우는 코드를 작성해보세요.

 

 

 

 

# n, m 입력받기
n, m = tuple(map(int, input().split()))

# 배열 선언
arr_2d = [
    [0 for _ in range(m)]
    for _ in range(n)
]

# 1씩 더해질 수 num 변수 선언
num = 0

for col in range(m):
    if col % 2 == 0: # 짝수 열
        for row in range(n): 
            arr_2d[row][col] = num 
            num += 1
    else: # 홀수 열
        for row in range(n-1, -1, -1): # 홀수 열에서는 거꾸로 할당되기 때문
            arr_2d[row][col] = num
            num += 1

# 출력
for row in range(n):
    for col in range(m):
        print(arr_2d[row][col], end = " ")
    print()