백준 [10810] 공 넣기 - 자바 풀이

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

10810번: 공 넣기

 

---

 

import java.util.Scanner;

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];
		
		for(int i=0; i<N; i++) {
			arr[i] = 0;
		}
		for(int i=0; i<M; i++) {
			int a = sc.nextInt();
			int b = sc.nextInt();
			int c = sc.nextInt();
			
			for(int j=0; j<b-a+1; j++) {
				arr[a+j-1] = c;
			}
			
		}
		for(int i=0; i<N; i++) {
			System.out.print(arr[i] + " ");
		}
	}
}

간단한 문제이다.

문제는 읽다보면 복잡한데, 예시를 읽고 이해하면 된다.

N개의 0이 주어지고, M번 조건을 반복한다.

I번에서 J번까지의 숫자를 K로 바꾼다.

 

int N = sc.nextInt();
int M = sc.nextInt();
int[] arr = new int[N];

N 과 M을 정수 타입으로 입력받는다. 

그리고 정수 타입의 arr 배열을 선언한다.

 

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

for문을 사용하여 arr배열에 N개의 0을 입력한다.

 

for(int i=0; i<M; i++) {
	int a = sc.nextInt();
	int b = sc.nextInt();
	int c = sc.nextInt();
	
	for(int j=0; j<b-a+1; j++) {
		arr[a+j-1] = c;
	}

a, b, c 를 정수 타입으로 나란히 입력받는다.

for문을 사용하여 M번 반복하게 만들고, 

변수 j가 b-a+1번 반복하는 for문을 만든다. 

 

왜냐! 예를 들어 1~3이 다른 수로 바뀌어야 한다면, 3번 반복해야 하는데 이걸 a, b와 연관된 식으로 만들면 b-a+1이 된다.

예시로 3-1+1이니 3번 반복하는 것이 맞다.

 

몇 번째 수부터 c로 바뀌어야 하나 생각해보면, 이 식은 j와 연관되어야 증가할 것이다. 

그러므로 위의 예시로 1~3번째 수를 2로 바꾼다 하면 a+j-1이 되면 된다.

예시로 1+0-1은 0번째 수가 된다. arr 배열에서는 맨 앞 수는 0번째부터 시작하니 말이다.

 

for(int i=0; i<N; i++) {
	System.out.print(arr[i] + " ");
}

마지막으로 for문을 사용해서 arr 배열에 있는 수를 배열하면 끝난다.

 

기분 좋은 정답!