문제 설명
부모님과 함께 놀러간 영일이는
설탕과자(설탕을 녹여 물고기 등의 모양을 만든 것) 뽑기를 보게 되었다.
길이가 다른 몇 개의 막대를 바둑판과 같은 격자판에 놓는데,
막대에 있는 설탕과자 이름 아래에 있는 번호를 뽑으면 설탕과자를 가져가는 게임이었다.
(잉어, 붕어, 용 등 여러 가지가 적혀있다.)
격자판의 세로(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
My code
h,w = map(int,input().split())
n = int(input())
arr=[]
for i in range(h):
arr.append([])
for j in range(w):
arr[i].append(0)
for i in range(n):
l,d,x,y = map(int,input().split())
for j in range(l):
if d==0:
arr[x-1][y+j-1]=1
else:
arr[x+j-1][y-1]=1
for i in range(h):
for j in range(w):
print(arr[i][j],end=' ')
print()
Answer
h,w = input().split()
h = int(h)
w = int(w)
m = []
for i in range(h+1) :
m.append([])
for j in range(w+1) :
m[i].append(0)
n = int(input())
for i in range(n) :
l,d,x,y = input().split()
if int(d)==0 :
for j in range(int(l)) :
m[int(x)][int(y)+j] = 1
else :
for j in range(int(l)) :
m[int(x)+j][int(y)] = 1
for i in range(1, h+1) :
for j in range(1, w+1) :
print(m[i][j], end=' ')
print()
Code Review
l, d, x, y 가 입력 되었을 때 만약 d가 0이라면 가로로 막대를 놓아야 합니다. 따라서 길이 l 만큼 arr[x][y+j] 즉, y 의 위치에 j 를 더한 위치 모두 1을 넣어줍니다. 만약 d가 1이라면 세로로 막대를 놓아야 합니다. 따라서 길이 l 만큼 arr[x+j][y] 즉, x의 위치에 j 를 더한 위치 모두 1을 넣어줍니다. 마지막으로 arr 를 출력하면 막대가 놓인 위치를 출력 할 수 있습니다.
댓글