[코드업] 6097번 설탕과자 뽑기

문제

길이가 다른 몇 개의 막대를 바둑판과 같은 격자판에 놓는데, 막대에 있는 설탕과자 이름 아래에 있는 번호를 뽑으면 설탕과자를 가져가는 게임이 있다. (잉어, 붕어, 용 등 여러 가지가 적혀있다.)

격자판의 세로(h), 가로(w), 막대의 개수(n), 각 막대의 길이(l), 막대를 놓는 방향(d:가로는 0, 세로는 1)과 막대를 놓는 막대의 가장 왼쪽 또는 위쪽의 위치(x, y)가 주어질 때, 격자판을 채운 막대의 모양을 출력하는 프로그램을 만들어보자.

 

입력

첫 줄에 격자판의 세로(h), 가로(w) 가 공백을 두고 입력되고,
두 번째 줄에 놓을 수 있는 막대의 개수(n)
세 번째 줄부터 각 막대의 길이(l), 방향(d), 좌표(x, y)가 입력된다.
1 <= w, h <= 100
1 <= n <= 10
d = 0 or 1
1 <= x <= 100-h
1 <= y <= 100-w

 

출력

모든 막대를 놓은 격자판의 상태를 출력한다.
막대에 의해 가려진 경우 1, 아닌 경우 0으로 출력한다.
단, 각 숫자는 공백으로 구분하여 출력한다.

 

입력 예시 

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

 

---

 

나의 풀이

 

1. 빈칸 리스트 만들기

h, w = map(int, input().split())
ls = [[0 for j in range(w)] for i in range(h)]

 

2. 가로 혹은 세로 방향으로(d) l만큼 이동하면서 1로 수정하기

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

- 입력받은 x, y 좌표는 1~n까지로 되어있고, 리스트의 인덱스는 0~n-1이기 때문에 우리가 원하는 좌표 탐색을 위해서는 -1을 해줘야 한다.

 

3. 출력 예시와 동일한 방식으로 출력하기

for a in range(h):
    for b in range(w):
        print(ls[a][b], end=' ')
    print('')