#10810 : 공 넣기

< 문제 내용 >

• 도현이는 바구니를 총 N개 가지고 있고, 각각의 바구니에는 1번부터 N번까지 번호가 매겨져 있다. 또, 1번부터 N번까지 번호가 적혀있는 공을 매우 많이 가지고 있다. 가장 처음 바구니에는 공이 들어있지 않으며, 바구니에는 공을 1개만 넣을 수 있다.
  도현이는 앞으로 M번 공을 넣으려고 한다. 도현이는 한 번 공을 넣을 때, 공을 넣을 바구니 범위를 정하고, 정한 바구니에 모두 같은 번호가 적혀있는 공을 넣는다. 만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다.
  

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

   

 

< 입력 조건 >

• 첫째 줄에 N과 M이 주어진다.
  둘째 줄부터 M개의 줄에 걸쳐서 공을 넣는 방법이 주어진다. 각 방법은 세 정수 i, j, k로 이루어져 있으며, i번 바구니부터 j번 바구니까지에 k번 번호가 적혀 있는 공을 넣는다는 뜻이다. 예를 들어, 2 5 6은 2번 바구니부터 5번 바구니까지에 6번 공을 넣는다는 뜻이다.
  도현이는 입력으로 주어진 순서대로 공을 넣는다. 

 

 

< 출력 형태 >

• 1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다.

 

 

< 풀이 및 회고 >

(풀이 코드)

 

N, M = map(int, input().split())  # N개의 바구니, M번 공을 넣을 예정

bucket = [0]*N     # 0으로 초기화된 N개의 바구니 생성

for a in range(M):		# M번 반복

    i, j, k = map(int, input().split())      
    # i부터 j까지 k번호의 공을 넣겠다!(하나씩 넣어야 함)

    for b in range(i-1, j):     # b는 횟수
    # i부터 j까지 반복
        bucket[b] = k   # 인덱스 i부터 j까지의 위치에 k값 삽입

print(' '.join(map(str, bucket)))

 

• 제일 먼저 문제의 원리를 파악하는 것이 중요했던 문제였다.
•  여기서 다소 어려웠던 부분은 i부터 j까지 반복해서 '앞에서부터' k를 집어넣는 것이었다.
  • for b in range(i-1, j):
            bucket[b] = k
    
    이 코드는 파이썬이 배열의 인덱스가 0부터 시작하는 0-based 인덱스이지만 현재 입력으로 받는 i와 j는 1부터 시작하는 1-based 인덱스 이기 때문에 i-1부터 시작해서 j-1까지의 범위로 지정함으로써 0-based 인덱스로 변환시켜 준 것이다.
• join 메서드를 사용하게 되면 '리스트의 요소들을 원하는 형식으로 연결하여 출력'할 수 있다.
  • print(' '.join(map(str, bucket)))
    
    해당 코드는 bucket 리스트의 모든 요소들을 문자열로 변환 후, 공백으로 연결해서 하나의 문자열로 출력하는 코드이다.