문제
미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.
공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.
1초 동안 아래 적힌 일이 순서대로 일어난다.
- 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
- (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
- 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
- 확산되는 양은 Ar,c/5이고 소수점은 버린다.
- (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
- 공기청정기가 작동한다.
- 공기청정기에서는 바람이 나온다.
- 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
- 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
- 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.
다음은 확산의 예시이다.
왼쪽과 오른쪽에 칸이 없기 때문에, 두 방향으로만 확산이 일어났다.
인접한 네 방향으로 모두 확산이 일어난다.
공기청정기가 있는 칸으로는 확산이 일어나지 않는다.
공기청정기의 바람은 다음과 같은 방향으로 순환한다.
방의 정보가 주어졌을 때, T초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.
입력
첫째 줄에 R, C, T (6 ≤ R, C ≤ 50, 1 ≤ T ≤ 1,000) 가 주어진다.
둘째 줄부터 R개의 줄에 Ar,c (-1 ≤ Ar,c ≤ 1,000)가 주어진다. 공기청정기가 설치된 곳은 Ar,c가 -1이고, 나머지 값은 미세먼지의 양이다. -1은 2번 위아래로 붙어져 있고, 가장 윗 행, 아랫 행과 두 칸이상 떨어져 있다.
출력
첫째 줄에 T초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.
예제 입력
7 8 1
0 0 0 0 0 0 0 9
0 0 0 0 3 0 0 8
-1 0 5 0 0 0 22 0
-1 8 0 0 0 0 0 0
0 0 0 0 0 10 43 0
0 0 5 0 15 0 0 0
0 0 40 0 0 0 20 0예제 출력
188코드
import java.io.*;
import java.util.StringTokenizer;
public class Main {
public static int[][] map = new int[51][51];
public static int[] dr = {-1, 0, 1, 0};
public static int[] dc = {0, -1, 0, 1};
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 R = Integer.parseInt(st.nextToken());
int C = Integer.parseInt(st.nextToken());
int T = Integer.parseInt(st.nextToken());
int[] air_r = new int[2];
int air_cnt = 0;
for (int i = 1; i <= R; i++) {
st = new StringTokenizer(br.readLine());
for (int j = 1; j <= C; j++) {
map[i][j] = Integer.parseInt(st.nextToken());
if (map[i][j] == -1) { // 공기청정기가 설치된 장소 저장
air_r[air_cnt++] = i;
}
}
}
while (T > 0) {
int[][] copy_map = new int[51][51];
for (int i = 1; i <= R; i++) {
for (int j = 1; j <= C; j++) {
if (map[i][j] <= 0) { // 미세먼지가 없거나 공기청정기가 설치된 곳이라면 건너뜀
if (map[i][j] == -1) copy_map[i][j] = map[i][j];
continue;
}
int cnt = 0;
for (int k = 0; k < 4; k++) { // 1. 미세먼지가 확산된다.
int next_r = i + dr[k];
int next_c = j + dc[k];
if (next_r > 0 && next_c > 0 && next_r <= R && next_c <= C) {
if (map[next_r][next_c] != -1) { // 인접한 공기청정기가 없는 경우에만
copy_map[next_r][next_c] += map[i][j] / 5;
cnt++;
}
}
}
copy_map[i][j] += map[i][j] - (map[i][j] / 5) * cnt;
}
}
map = copy_map;
// 2. 공기청정기가 작동한다.
for (int i = air_r[0] - 1; i > 1; i--) { // 내려오는 방향
map[i][1] = map[i - 1][1];
}
for (int i = 1; i < C; i++) { // <-
map[1][i] = map[1][i + 1];
}
for (int i = 1; i < air_r[0]; i++) { // 올라오는 방향
map[i][C] = map[i + 1][C];
}
for (int i = C; i > 2; i--) { // ->
map[air_r[0]][i] = map[air_r[0]][i - 1];
}
map[air_r[0]][2] = 0;
for (int i = air_r[1] + 1; i < R; i++) {
map[i][1] = map[i + 1][1];
}
for (int i = 0; i < C; i++) {
map[R][i] = map[R][i + 1];
}
for (int i = R; i > air_r[1]; i--) {
map[i][C] = map[i - 1][C];
}
for (int i = C; i > 2; i--) {
map[air_r[1]][i] = map[air_r[1]][i - 1];
}
map[air_r[1]][2] = 0;
T--;
}
int sum = 0;
for (int i = 1; i <= R; i++) {
for (int j = 1; j <= C; j++) {
if (map[i][j] > 0) {
sum += map[i][j];
}
}
}
bw.write(sum + "\n");
br.close();
bw.flush();
bw.close();
}
}정리
- 공기청정기가 작동할 때 미세먼지를 일일이 옮겨주는 작업에서
i범위를 잘못 넣어 고생 좀 했다. - 처음에 어렵게 생각한 거와 다르게 풀고 나니 단순 구현 문제였다.
기억하기 위해 기록하는 개발로그👣