본문 바로가기

Python/백준

#2 [python] 백준 10810번 공넣기

728x90

 

 

 

Python Code

 

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 = ' ')

 

1. 바구니 갯수 N, 반복횟수 M 입력 받기

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

 

2. list basket 선언 후 값 0으로 초기화(공이 없는 바구니는 0 출력)

basket = [0 for _ in range(N)]

 

3. 반복횟수 M번동안 i,j,k를 입력받고 list 값을 갱신하기 위해 for 문 추가적으로 실행

for _ in range(M): i,j,k = map(int, input().split())

 

4. j 바구니 까지 값 갱신되어야 하므로 반복문 j+1 까지 / 바구니 n 은 0 번부터 존재하므로 n-1까지

for n in range(i,j+1):

     basket[n-1] = k

 

5. list 값을 공백으로 출력하기 위해 end= ' ' 넣어주기

for n in range(N):

   print(basket[n], end = '  ')

 

 

참고 

https://kevinitcoding.tistory.com/entry/%EB%B0%B1%EC%A4%80Python-10810%EB%B2%88-%EA%B3%B5-%EB%84%A3%EA%B8%B0-%EB%AC%B8%EC%A0%9C