백준 > 단계별로 풀어보기 > 1차원 배열 > 10810번: 공 넣기

문제

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

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

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

입력

첫째 줄에 N (1 ≤ N ≤ 100)과 M (1 ≤ M ≤ 100)이 주어진다.

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

도현이는 입력으로 주어진 순서대로 공을 넣는다.

출력

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

예제 입력 1 복사

5 4
1 2 3
3 4 4
1 4 1
2 2 2

예제 출력 1 복사

1 2 1 1 0

 

 

진짜 코딩테스트 같은 문제를 처음 만났다. 어떻게 해야할지 몰라서 그냥 검색을 해봤다.

import sys

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

for q in range(M):
    i, j, k = map(int, sys.stdin.readline().split())

난 이렇게만 작성했는데, 문제에서 "가장 처음 바구니에는 공이 들어있지 않다" 고 했으니 모든 바구니를 0으로 초기화한다고 한다. 컴프리헨션으로 작성했는데 [0]*N도 괜찮은 것 같다.

basket = [0 for i in range(N]

또한 출력 설명에서 1번 바구니부터 N번 바구니에 들어있는 공의 번호를 공백으로 구분해 출력한다. 공이 들어있지 않은 바구니는 0을 출력한다. 라고 되어 있으니 0으로 초기화하는게 맞는 것 같다.

 

import sys

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

for _ in range(M):
    i, j, k = map(int, sys.stdin.readline().split())
    for q in range(i, j+1): # i부터 j까지
        basket[q-1] = k    # 1번 바구니는 0번 인덱스, 2번 바구니는 1번 인덱스
for i in range(N):
    print(basket[i], end=" ")

# 이것도 가능
#for i in basket:
#	print(i, end=" ")

 

만약, 바구니에 공이 이미 있는 경우에는 들어있는 공을 빼고, 새로 공을 넣는다. 공을 넣을 바구니는 연속되어 있어야 한다. >>>>> basket[i-1] = k 

i, j, k는 i부터 j까지 k번의 공을 넣는다는 소리임그런데

for q in range(i, j+1):
	basket[q-1]=k

을 한 이유는,

1번 바구니 => 인덱스가 0

5번 바구니 => 인덱스가 4 이기 때문이다. i, j+1은 i부터 j까지 라는 말이다.

q는 1번 바구니~5번 바구니를 뜻하고, q-1는 그에 해당하는 인덱스를 뜻한다.

 

처음 보는 코딩테스트 다운 문제였고 너무 당황했다. 코딩테스트 문제에 대한 겁을 지우고 싶다.