[백준] 코딩테스트 문제 10810 : 공 넣기

https://www.acmicpc.net/problem/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.util.*;
public class Main{
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();  // 배열 크기 입력
        int M = sc.nextInt();  // 반복 횟수 입력
        int[] arr = new int[N+1];  // 배열 크기 설정(0번이 없으므로 1 더해줌)
        for(int n = 1; n <= N; n++){  // 처음 배열 값에 모두 0 입력
                arr[n] = 0;
        }    
        for(int q = 0; q < M; q++){
            int i = sc.nextInt();
            int j = sc.nextInt();
            int k = sc.nextInt();
            
            for(int l = i; l <= j; l++){  // i부터 j번까지의 배열에 k 값을 입력
                arr[l] = k;
            }
        }
        
        for(int n = 1; n <= N; n++){ // 배열 값 출력
            System.out.print(arr[n] + " ");
        }
        
    }
}

 배열 크기와 반복 횟수를 처음 입력받고, 입력받은 크기의 배열을 생성한다.(이때 입력받은 배열 크기에 1을 더하여 0번 배열을 사용하지 않도록 해줬음) 

 이후 처음엔 모든 배열 값에 0을 넣어주었다.이전에 제출한 답들은 k값을 입력한 뒤에 k값이 아닌 곳에 0을 넣으려 했는데 그렇게 되면 이전에 이미 값을 넣었지만 k값과 일치하지 않는 곳들도 모두 0이 되어버려서 초반에 0을 먼저 넣어주는 것으로 수정했다. 이후 i, j, k값을 입력받은 뒤 for문을 통해 i부터 j까지의 배열에 k값을 입력한 뒤 배열 값을 출력했다.