[P&B] #48 BAEKJOON 10810

백준 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을 출력한다.

 

나의 문제 풀이 코드

import java.io.*;
import java.util.StringTokenizer;

public class bj10810 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[] balls = new int[N];
        for (int i = 0; i < N; i++) {
            balls[i] = 0;
        }

        for (int i = 0; i < M; i++) {
            StringTokenizer st2 = new StringTokenizer(br.readLine());
            int a = Integer.parseInt(st2.nextToken());
            int b = Integer.parseInt(st2.nextToken());
            int c = Integer.parseInt(st2.nextToken());
            for (int j = a-1; j <= b-1; j++) {
                    balls[j] = c;
            }
        }

        for (int ball: balls) {
            bw.write(ball+" ");
        }
        bw.flush();
    }
}

 

문제 풀이 코멘트

길이가 N인 배열을 생성하고 초기값으로 0을 설정한 후, M번의 연산을 수행하여 배열의 일부를 변경한 뒤 최종 배열을 출력하는 프로그램입니다. 주로 '구간 갱신' 문제와 관련이 있습니다.

• 길이가 N인 balls 배열을 생성하고, 모든 요소를 초기값 0으로 설정합니다.
• 다음 M개의 줄에 대해서 반복문을 사용하여 연산을 수행합니다.
• 각 연산은 세 개의 정수 a, b, c를 입력받아 처리합니다. 이 연산은 배열 balls의 인덱스 a부터 b까지의 구간을 c로 갱신하는 것을 의미합니다. 따라서 a-1부터 b-1까지의 인덱스를 사용하여 해당 구간의 값을 c로 변경합니다.