코드업[CodeUp] (파이썬) 6097번 풀이

Python으로 구현한 6097번 설탕과자 뽑기 문제 풀이입니다.

 

https://codeup.kr/problem.php?id=6097 

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

---

h, w = map(int, input().split())

p = [[0] * w for _ in range(h)]

n = int(input())

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

for i in range(h) :
  for j in range(w) :
    print(p[i][j], end = ' ')
  print()

 

격자판의 세로(h)와 가로(w)를 공백으로 구분하여 입력받고 입력받은 크기의 격자판의 각 값을 0으로 초기화시켜준다.

 

또한, 줄에 놓을 수 있는 막대의 개수(n)을 입력받은 후 n번만큼 반복문을 수행하여 그 내부에서는

 

각 막대의 길이(l)와 방향(d), 좌표(x, y)를 입력받도록 한다. 입력을 받은 후에는 입력받은 막대의 길이(l) 만큼 반복하여

 

방향이 0이면 가로 방향으로 막대를 하나씩 세우고, 반대로 방향이 1이라면 세로 방향으로 막대를 하나씩 세운다.

 

최종적으로 모든 막대를 놓은 격자판의 상태를 출력한다.