[백준] 10810번 - 공 넣기 [Java]

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

 

[백준] 10810번 - 공 넣기 [Java]

[백준] 10810번 - 공 넣기 [Java]

---

1.  아이디어

 

int형 배열은 0으로 초기화되는 점을 활용해 N+1 크기의 배열을 활용하면 간단하게 해결할 수 있다.

---

2. 문제풀이

 

인덱스와 바구니의 번호를 맞춘 int형 배열에 직접 공을 넣어서 구현했다. 새로운 값을 넣으면 기존 값에 덮어써지는 것이 공을 빼고 새로 넣는 것과 동일한 구조이다.

---

3. 코드

 

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

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

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[] arr = new int[1 + N];

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int start = Integer.parseInt(st.nextToken());
            int end = Integer.parseInt(st.nextToken());
            int num = Integer.parseInt(st.nextToken());

            for (int j = start; j <= end; j++) {
                arr[j] = num;
            }
        }

        for (int i = 1; i <= N; i++) {
            sb.append(arr[i]).append(" ");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기