[CodeUp] 기초100제 파이썬 문풀 6097 | 기초-리스트

 

 

6097 설탕과자 뽑기

h, w = map(int, input().split())
k =[[0 for i in range(w)]for j in range(h)]

n= int(input())
for i in range(n):
    l, d, x, y = map(int, input().split())
    if d==0:
        for j in range(l):
            k[x-1][y+j-1]=1
    else:
        for j in range(l):
            k[x+j-1][y-1]=1

for i in range(h):
    for j in range(w):
        print(k[i][j], end=' ')
    print()

 

1) 가로, 세로 값 받아서 리스트 만들기

2) 막대의 개수(테스트 케이스) 수 입력받아서 막대의 개수만큼 반복될 반복문 작성

3) 막대의 길이(l), 방향(d), 좌표(x, y) 값 받기

4) 방향에 따라 달라질 조건문 작성

4-1) 가로방향이면, 막대 길이만큼 반복하여 y+j 위치에 1씩 더하기

4-2) 세로 방향이면, 막대 길이만큼 반복하여 x+j 위치에 1씩 더하기

5) 리스트 출력

 

 

 

 

 

 

(x, y)에서 x가 가로 y가 세로인 줄 알았는데 아니더라고요 ㅠㅠ 계속 헤매다가 그림에서 x랑 y 위치보고 음?? 반대인가 하고 코드 바꾸니까 맞다고 뜨네요 ㅠㅠ 후 그림 없었으면 어쩔뻔....

 

 

아래 코드는 리스트 만드는 다양한 방법들 모아봤습니다

 

k =[[0 for i in range(w)]for j in range(h)]

k = [[0]*w for _ in range(h)]

k= []
for i in range(h):
    k.append([])
    for _ in range(w):
        k[i].append(0)

저는 첫 번째 방식을 사용 중입니다 두 번째도 간결하고 좋네요

 

 

 

 

 

 

 

 

 

틀린 내용 있다면 댓글로 알려주세요 :)