문제
미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.
공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.
1초 동안 아래 적힌 일이 순서대로 일어난다.
- 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
- 공기청정기가 작동한다.
공기청정기의 바람은 다음과 같은 방향으로 순환한다.
방의 정보가 주어졌을 때, 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번 위아래로 붙어져 있고, 가장 윗 행, 아랫 행과 두 칸이상 떨어져 있다.
코드
const fs = require("fs");
const input = fs.readFileSync("/dev/stdin", "utf-8").trim().split("\n");
const MapInfo = input[0].split(" ");
const r = parseInt(MapInfo[0]);
const c = parseInt(MapInfo[1]);
const goalTime = parseInt(MapInfo[2]);
const Map = input
.slice(1, input.length)
.map((e) => e.split(" ").map((n) => parseInt(n)));
const dx = [1, -1, 0, 0];
const dy = [0, 0, 1, -1];
const cleanerChecking = () => {
const cleaner = [];
for (let i = 0; i < r; i++) {
if (Map[i][0] === -1) {
cleaner.push(i);
}
}
return { upperGoalX: cleaner[0], lowerGoalX: cleaner[1] };
};
const check = (x, y, save, upperGoalX, lowerGoalX) => {
let cnt = 0;
let value;
for (let i = 0; i < 4; i++) {
const nx = x + dx[i];
const ny = y + dy[i];
if (nx < 0 || ny < 0 || nx >= r || ny >= c) {
continue;
}
if ((nx === upperGoalX && ny === 0) || (nx === lowerGoalX && ny === 0)) {
continue;
}
value = Math.floor(Map[x][y] / 5);
save[nx][ny] += value;
cnt += 1;
}
if (value || value === 0) {
save[x][y] += Map[x][y] - cnt * value;
} else {
save[x][y] += Map[x][y];
}
};
const cleanUp = (goalX) => {
for (let i = goalX - 1; i >= 0; i--) {
if (i + 1 !== goalX) {
Map[i + 1][0] = Map[i][0];
}
}
for (let i = 1; i < c; i++) {
Map[0][i - 1] = Map[0][i];
}
for (let i = 1; i <= goalX; i++) {
Map[i - 1][c - 1] = Map[i][c - 1];
}
for (let i = c - 2; i >= 0; i--) {
Map[goalX][i + 1] = Map[goalX][i];
}
};
const cleanDown = (goalX) => {
for (let i = goalX + 1; i < r; i++) {
if (i - 1 !== goalX) {
Map[i - 1][0] = Map[i][0];
}
}
for (let i = 1; i < c; i++) {
Map[r - 1][i - 1] = Map[r - 1][i];
}
for (let i = r - 2; i >= goalX; i--) {
Map[i + 1][c - 1] = Map[i][c - 1];
}
for (let i = c - 2; i >= 0; i--) {
Map[goalX][i + 1] = Map[goalX][i];
}
};
const mapChecking = (save, upperGoalX, lowerGoalX) => {
const searchArr = [];
for (let i = 0; i < r; i++) {
for (let j = 0; j < c; j++) {
if (Map[i][j] > 0) {
searchArr.push([i, j]);
}
}
}
searchArr.forEach((point) =>
check(point[0], point[1], save, upperGoalX, lowerGoalX)
);
};
const setMap = (save) => {
for (let i = 0; i < r; i++) {
for (let j = 0; j < c; j++) {
Map[i][j] = save[i][j];
}
}
};
const MapCountingCheck = () => {
let cnt = 0;
for (let i = 0; i < r; i++) {
for (let j = 0; j < c; j++) {
if (Map[i][j] > 0) {
cnt += Map[i][j];
}
}
}
return cnt;
};
const solution = () => {
const { upperGoalX, lowerGoalX } = cleanerChecking();
for (let i = 0; i < goalTime; i++) {
const save = Array.from(Array(r), () => new Array(c).fill(0));
mapChecking(save, upperGoalX, lowerGoalX);
setMap(save);
cleanUp(upperGoalX);
cleanDown(lowerGoalX);
}
console.log(MapCountingCheck());
};
solution();
먼저 1번과정인 확산은 mapChecking을 이용하여 0보다 클때 searchArr에 넣어서 한꺼번에 check해주었다. 하나하나 바로 확산시켜버리면 동시에 확산한다는 조건에 어긋나서 틀리게 된다.
확산도 역시 바로 Map을 변형하는게 아니라 save배열에 저장한뒤 마지막에 한꺼번에 setMap을 이용하여 Map을 변경해준다.
그 다음 2번과정은 문제에서 주어진대로 하나하나 직접 구현하였다.
