[백준] 10810번 파이썬 python(공 넣기)

▷ 문제

도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.

도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.

공을 어떻게 넣을지가 주어졌을 때, M번 공을 넣은 이후에 각 바구니에 어떤 공이 들어 있는지 구하는 프로그램을 작성하시오.

 

▷ 전체 코드

N, M = map(int, input().split())

X = []
for i in range(N):
    X.append(0)

for b in range(M):
    i, j, k = map(int, input().split())
    for b in range(i, j+1):
        X[b-1] = k

for i in range(N):
    print(X[i], end=" ")

 

▷ 코드 분석

N, M = map(int, input().split())

변수 N에 바구니의 개수를 입력하고, 변수 M에 공을 넣을 횟수를 입력한다.map() 함수를 사용해 지정된 함수를 입력 받은 요소에 각각 적용시킨다.input() 함수를 사용해 요소들을 문자열로 입력 받는다. 그 후 입력 받은 요소들을 split() 함수를 사용해 공백을 기준으로 나눠준다.int() 함수를 사용해 문자열을 정수형으로 바꿔준다.

 

X = []

리스트 X를 만들어준다.

 

for i in range(N):
    X.append(0)

for 문을 사용해 N번 반복하는 반복문을 만들어준다.

.append() 함수를 사용해 리스트 X에 0을 추가한다.

 

for 문을 사용했으므로 리스트 X에 0이 N개 추가된다.

 

for b in range(M):
    i, j, k = map(int, input().split())
    for b in range(i, j+1):
        X[b-1] = k

i번부터 j번까지의 바구니에 k번 번호가 적혀져 있는 공을 넣어준다.

 

 

for 문을 사용해 M번 반복하는 반복문을 만들어준다.

i, j, k를 입력 받은 후, for 문을 사용해 i번부터 j번까지 반복하는 반복문을 만들어준다.

리스트 X에 i번부터 j번까지의 요소를 k로 교체한다.

 

for i in range(N):
    print(X[i], end=" ")

for 문을 사용해 N번 반복하는 반복문을 만들어준다.

리스트 X에 각 요소를 공백을 기준으로 나눠서 출력한다.