[빡공팟 5기] 6097 : [기초-리스트] 설탕과자 뽑기(py)

문제 링크 : https://codeup.kr/problem.php?id=6097 

[기초-리스트] 설탕과자 뽑기(py)

---

📋 문제 설명

격자판의 세로(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

 

 

 

✏️ 풀이과정

h, w = map(int, input().split())  #격자판의 세로, 가로 입력
n = int(input())  #막대의 개수 입력
inf = [[0 for i in range(4)] for j in range(n)]  #막대의 길이, 방향, 좌표를 저장할 리스트
prt = [[0 for i in range(w)] for j in range(h)]  #막대를 놓을 격자판을 저장할 리스트

for i in range(n):
    inf[i] = input().split()  #막대 정보 입력
    for j in range(4):
        inf[i][j] = int(inf[i][j])  #int형으로 변환
    inf[i][2] -= 1
    inf[i][3] -= 1  #인덱스 값으로 사용하기 위해 -1을 해줌
        
for i in range(n):
    if inf[i][1] == 0:  #0이면 왼->오 순으로 가로로 막대를 놓음
        for j in range(inf[i][0]):
            prt[inf[i][2]][inf[i][3]+j] = 1  #막대를 놓은 곳은 1로 변경
    else:  #1이면 위->아래 순으로 세로로 막대를 놓음
        for j in range(inf[i][0]):
            prt[inf[i][2]+j][inf[i][3]] = 1  #막대를 놓은 곳은 1로 변경
            
for i in range(h): 
    for j in range(w):
        print(prt[i][j], end = " ")
    print()

이번 문제는 문제를 잘 읽고 이해한다면 어렵지 않게 풀 수 있는 문제였던 것 같다.

이런 문제를 풀 때에 어떻게 변수를 만들고 입력을 받을지, 그 변수들을 어떻게 활용할지 고민이 된다.

주어지는 조건들이 어렵게 되어있지 않아서 나름 금방 푼 문제였던 것 같다.