코드업 기초 100제 - 1098 : [기초-2차원배열] 설탕과자 뽑기

1. 문제

---

• 언어 : python3

부모님과 함께 유원지에 놀러간 영일이는
설탕과자(설탕을 녹여 물고기 등의 모양을 만든 것) 뽑기를 보게 되었다.

길이가 다른 몇 개의 막대를 바둑판과 같은 격자판에 놓는데,

막대에 있는 설탕과자 이름 아래에 있는 번호를 뽑으면 설탕과자를 가져가는 게임이었다.
격자판의 세로(h), 가로(w), 막대의 개수(n), 각 막대의 길이(l),
막대를 놓는 방향(d:가로는 0, 세로는 1)과
막대를 놓는 막대의 가장 왼쪽 또는 위쪽의 위치(x, y)가 주어질 때,

격자판을 채운 막대의 모양을 출력하는 프로그램을 만들어보자.

 

입력

첫 줄에 격자판의 세로(h), 가로(w) 가 공백을 두고 입력되고,
두 번째 줄에 놓을 수 있는 막대의 개수(n)
세 번째 줄부터 각 막대의 길이(l), 방향(d), 좌표(x, y)가 입력된다.

출력

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

 

2. 풀이 결과(소스코드)

---

h, w = input("판의 세로 크기, 가로 크기 입력 : ").split()
h, w = int(h), int(w)
arrary = []
n = int(input("막대의 개수 입력 :"))

# 입력받은 크기만큼 이중 for문을 돌려서 격자판 생성
for i in range(h):
    arrary.append([])
    for j in range(w):
        arrary[i].append(0)


for i in range(n):
    l, d, x, y = input("막대의 길이, 방향(0은 가로, 1은 세로), x좌표, y좌표 입력 : ").split()
    l, d, x, y = int(l), int(d), int(x), int(y)
    # 막대의 길이만큼 반복문 진행
    for j in range(l):
        # 막대가 가로로 놓인 경우, 입력받은 크기만큼 가로(x)축 방향으로 이동하면서 배열의 값을 1로 바꿈
        if int(d) == 0:
            arrary[x-1][y-1+j] = 1
        # 그외의 경우(세로)인, 입력받은 크기만큼 세로(y)축 방향으로 이동하면서 배열의 값을 1로 바꿈
        else:
            arrary[x-1+j][y-1] = 1

for i in range(h):  # 막대가 놓여진 최종 결과값 출력
    for j in range(w):
        print(arrary[i][j], end=' ')
    print('')

 

3. 회고

---

막대의 진행방향에 따라 분기해서 x, y 좌표에 1을 넣어야했는데 이 부분에서 헤맸다.

결국 구글링해서 다른 분의 풀이를 보고 진행... 좌표에 대한 이해가 필요하다.