[code up] 파이썬 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  (d = 0 가로 이동 / d =1 세로 이동)
1 <= x <= 100-h
1 <= y <= 100-w

---

출력

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

코드

w, h = map(int, input().split())
li = []
for i in range(w):
    li.append([])
    for j in range(h):
        li[i].append(0)
n = int(input())
for k in range(n):
    l, d, x, y = map(int, input().split())
    for h in range(l):
        if d == 0 :
            li[x-1][y-1] = 1
            y += 1
        else :
            li[x-1][y-1] =1
            x +=1
for i in li :
    print(" ".join(map(str, i)))

 

- map(function, iterable)함수를 이용하여 두 개의 입력 값을 바로 정수형 데이터로 바꾸어준다. (split() 함수는 문자열을 일정한 규칙으로 잘라서 리스트로 만들어 주는 함수)

li 리스트에 li.append([]) 빈 2중 리스트를 만들어준다. for j in range(h): li[i].append(0) 을 통해 모든 행열에 0을 넣어준다.

d의 값에 따라 가로, 세로로 이동하며 0을 1로 바꿔준다. 가로 이동일 경우 y+=1 / 세로 이동일 경우 x+=1

"구분자".join()함수를 통해 문자열을 합쳐준다.