[CodeUp] 기초 100제 1098번 2차원 배열 설탕과자 뽑기 풀이 (Python)

파이썬 문법에 익숙해지기 위해 CodeUp 기초 100제를 풀어보았습니다.

 

문제

[기초-2차원배열] 설탕과자 뽑기

[기초-2차원배열] 설탕과자 뽑기

격자판의 세로(h), 가로(w), 막대의 개수(n), 각 막대의 길이(l),
막대를 놓는 방향(d:가로는 0, 세로는 1)과
막대를 놓는 막대의 가장 왼쪽 또는 위쪽의 위치(x, y)가 주어질 때,

격자판을 채운 막대의 모양을 출력하는 프로그램을 만들어보자.

 

입력 예시	출력 예시
5 5 3 2 0 1 1 3 1 2 3 4 1 2 5	1 1 0 0 0 0 0 1 0 1 0 0 1 0 1 0 0 1 0 1 0 0 0 0 1

 

풀이

h, w = map(int, input().split())
array = [[0] * w for _ in range(h)]

n = int(input())
for _ in range(n):
    l, d, x, y = map(int, input().split())
    for i in range(l):
        if d == 0:
            array[x - 1][y - 1 + i] = 1
        else:
            array[x - 1 + i][y - 1] = 1

for x in range(h):
    for y in range(w):
        print(array[x][y], end=' ')
    print('')