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

문제

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

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

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

 

정답코드 

N, M = map(int, input().split())
basket = [0] * (N+1)

for _ in range(M):
    i, j, k = map(int, input().split())
    for n in range(i, j+1):
        basket[n] = k 

for i in range(1, N+1):
    print(basket[i], end = ' ')

 

풀이 

1. N과 M을 입력받아 바구니의 개수와 공을 넣을 횟수를 지정해 준다. 
2. basket이라는 리스트를 선언한다. 
3. for 문으로 공을 N번 넣을 구조를 구현하고
4. 공을 넣을 형식(i번째 바구니부터 j번째 바구니까지, 번호 k의 공을 넣을 거다)을 지정해 준다. 
5. i번째부터 j번째까지 n반복문을 만들고
6. 차례대로 basket의 n번째 자리에 번호 k의 공을 넣어준다. 
7. 물론 반복문이 돌면서 번호 k는 계속 덮여씌워진다. 
8. 마지막으로 바구니 안에 있는 공들의 숫자를 출력한다. 

 

느낀 점 

이건 이해하는 게 5할이다. 

한번 이해하면 일단 여러 개의 요소를 담을 리스트를 구현해주어야 하는데, 리스트 초기화를 하는 방법을 배웠다. 

아직 리스트 안에 값을 넣는 것이 익숙하지 않은 것 같다. 

인덱싱은 0부터 시작해서 n-1까지라는 것을 잊지 말자.