코드업 파이썬 6097번 예제 : 설탕과자 뽑기

코드업 6097번 예제인 설탕과자 뽑기는 원하는 사이즈의 판을 생성해서 그 위에 막대를 놓는 프로그램을 만드는 문제입니다.

​

작동 순서

1. 가로 세로 크기를 입력받는다.

2. 입력받은 크기만큼 판을 생성한다.

3. 막대의 개수를 입력받는다.

4. 막대길이, 방향, 좌표를 입력받는다.

5. 입력받은 정보들을 바탕으로 막대를 놓는다.

6. 4,5번을 3번에서 입력받은 막대의 개수만큼 반복한다.

7. 막대가 놓인 판을 출력한다.

​

소스코드

h,w=map(int,input("가로 세로를 입력하세요.").split())#바둑판 크기입력
d=[[0]*w for i in range(h)]#바둑판 생성

n=int(input("막대의 개수를 입력하세요."))#막대개수 입력
for i in range(n):
    l,dir,x,y=map(int,input("막대의 길이, 방향, 놓을 좌표를 입력하세요.").split())#막대길이, 방향, 좌표 입력
    for j in range(l):#입력받은 좌표에서부터 막대길이 만큼 방향에 맞춰서 1로 변횐
        d[x-1][y-1]=1
        if dir==0:
            y+=1
        else:
            x+=1

print("판을 출력합니다.")
for i in range(h):#바둑판 출력
    for j in range(w):
        print(d[i][j],end=' ')
    print()