[python][CodeUp]Python 기초 100제 - 97번

문제

입력

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

ex. 

5 5

3

2 0 1 1

3 1 2 3

4 1 2 5

출력

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

ex. 

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 = input().split()
lst = []
for i in range(int(h)):
    lst.append([])
    for j in range(int(w)):
        lst[i].append(0)

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

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

우선 h와 w의 입력값을 받아준다.

해당 값을 가지고 for문을 사용해 2차원 리스트를 생성해준다.

 

l, d, x, y의 값을 입력받는다.

d의 여부에 따라 x에 l의 값을 더해야할지, y에 l의 값을 더해야할지 결정된다.

 

여기서 주의해야 할 점은 좌표와 리스트의 번호가 다르다는 것이다.

좌표에서는 1, 1 이지만, 리스트에서는 0, 0 이다.

 

그렇게 조건문을 통해 필요한 부분의 값을 1로 변경해준 뒤

이중 for문을 사용해 해당 값들을 보기 좋게 출력시킨다.

 

 

주의할 것!

h와 w가 헷갈렸고, x와 y의 방향도 헷갈렸음