코드업[CodeUp] Python 기초 100제 [6097번]

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

 

정답 및 가능한 정답

a,b=list(map(int,input().split()))

sugarmap=[]
for i in range(a):
    line=[]
    for i in range(b):
        line.append(0)
    sugarmap.append(line)
    
n=int(input())
for i in range(n):
    l,d,x,y=map(int,input().split())
    x=x-1
    y=y-1
    for j in range(l):
        if d==0:
            sugarmap[x][y+j]=1
        
        if d==1:
            sugarmap[x+j][y]=1
              
for i in range(a):
    for j in range(b):
        print(sugarmap[i][j],end=" ")
    print()

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()

 

해설

첫 줄에 격자판의 세로(h), 가로(w) 가 공백을 두고 입력한다. 
a,b=list(map(int,input().split()))

입력받은 세로(h)와 가로(w)로 2차원 리스트로 격자판을 생성한다. 
sugarmap=[]
for i in range(a):
    line=[]
    for i in range(b):
        line.append(0)
    sugarmap.append(line)

막대의 개수(n)을 입력받는다. 
n=int(input())

세 번째 줄부터 각 막대의 길이(l), 방향(d), 좌표(x, y)가 입력된다.
입력받은 좌표에서부터 정해진 방향에 따라 길이만큼의 좌표를 1으로 바꾼다. 
for i in range(n):
    l,d,x,y=map(int,input().split())
    x=x-1
    y=y-1
    for j in range(l):
        if d==0:
            sugarmap[x][y+j]=1
        if d==1:
            sugarmap[x+j][y]=1

그렇게 만들어진 2차원 리스트 격자판을 출력한다. 
for i in range(a):
    for j in range(b):
        print(sugarmap[i][j],end=" ")
    print()