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())       #막대의 개수

board = [0] * h
for i in range(len(board)):     #격자판 만들기
    board[i] = [0] * w

for i in range(n):  #막대의 개수만큼 반복문 실행
    l, d, x, y = map(int, input().split())  # 막대의길이, 방향, 좌표 입력받기

    if d == 0:  #막대의 방향이 가로
        for l in range(l):
            board[x-1][y-1+l] = 1

    if d == 1:  #막대의 방향이 세로
        for l in range(l):
            board[x-1+l][y-1] = 1


#board 출력
for i in range(len(board)):
    for j in range(len(board[i])):
        print(board[i][j], end=" ")
    print()

---

해설

 

입력받은 격자판의 세로, 가로 크기로 이차원 리스트 board를 생성 

반복문은 막대의 개수 만큼 실행

반복문 내용 

   1. l, d, x, y 입력받음 (막대의 길이, 방향, 좌표 )

   2. 두 가지 경우로 나눔 (막대의 방향이 세로인지, 가로인지)

   3. 막대의 방향이 가로이면 행은 그대로, 열 값만 증가 > 1 대입 (막대 길이 만큼 반복)

   4. 막대의 방향이 세로이면 행 값 증가, 열은 그대로 > 1 대입 (막대 길이 만큼 반복)

 

마지막으로 완성된 이차원 리스트 board 출력 

 

-참고-

좌표는 (0, 0) 이 아닌 (1, 1) 부터 시작하므로 board의 인덱스를 계산할 때 -1 을 해주었다.

나는 if 문 안에 for문을 작성했는데 다른 분 코드를 보니, for문이 동일하므로 for문 내에 if문을 작성해도 좋을 것 같다.

 

for l in range(l):

  if d == 0:

       ...

  if d == 1:

       ...