[백준/Python] 10810번 공 넣기 문제

■ 10810번 공 넣기 문제

출처: 백준 10810번 공 넣기 문제

 

■ 코드 풀이

 

아이디어는 다음과 같습니다.

1) 바구니 개수 N과 반복 횟수 M을 입력 받습니다.

2) basket이라는 list를 선언하고 값을 모두 0으로 초기화합니다. (공이 없는 바구니는 0을 출력해야 하므로)

3) M번 동안, 각각 i,j,k를 입력 받은 뒤 list의 값을 갱신하기 위해 for loop를 추가적으로 실행합니다.

4) 이 때, j 바구니 "까지" 값이 갱신되어야 하므로 반복문의 range는 (i, j+1)로 해줍니다.

5) 또한, 문제의 정의에 의해 바구니는 '1번'부터 존재하지만 list의 index는 '0번'부터 존재하므로 list의 값을 갱신할 때는 index에 -1을 빼서 업데이트 해줍니다.

5) list 값을 공백으로 출력하기 위해 end 옵션을 넣어 줍니다.

N, M = map(int, input().split())
basket = [0 for _ in range(N)]

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

for n in range(N):
    print(basket[n], end = ' ')