[JAVA]백준 알고리즘 10810번 : 공 넣기

https://www.acmicpc.net/problem/10810

10810번: 공 넣기

문제

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

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

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

 

풀이

문제의 조건에 따라 바구니에 공을 넣은 후, 마지막에 바구니에 들어있는 공의 번호들을 출력하는 문제이다.

 

바구니에 공이 이미 있는 경우 들어있는 공을 빼고, 새로 공을 넣는다고 되어있는데, 이 조건은 생각하지 않아도 된다.

배열 내에 값이 존재할 때, 다른 값을 넣으면 덮어씌워지기 때문이다.

 

바구니의 번호는 1번부터 N번까지 사용하므로, 편의를 위해 배열의 크기를 N + 1로 설정하였다.

 

코드

BufferedReader 클래스와 StringBuilder 클래스를 이용한 코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();

        st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[] basket = new int[N + 1]; // 바구니 배열

        for(int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int I = Integer.parseInt(st.nextToken()); // I번 바구니부터
            int J = Integer.parseInt(st.nextToken()); // J번 바구니까지에
            int K = Integer.parseInt(st.nextToken()); // K번 번호가 적혀있는 공을 넣는다.

            // 위 조건을 수행하는 반복문
            for(int j = I; j <= J; j++) {
                basket[j] = K;
            }
        }
        for(int i = 1; i <= N; i++) {
            sb.append(basket[i]).append(" ");
        }
        System.out.println(sb);
    }
}

 

틀린 부분이 있다면 정정해 주시면 감사하겠습니다.
궁금한 부분이 있거나, 다른 아이디어가 있으시면 자유롭게 댓글 남겨주세요!