[BOJ] 17144 미세먼지 안녕! (C/C++)

17144번: 미세먼지 안녕!

문제

미세먼지를 제거하기 위해 구사과는 공기청정기를 설치하려고 한다. 공기청정기의 성능을 테스트하기 위해 구사과는 집을 크기가 R×C인 격자판으로 나타냈고, 1×1 크기의 칸으로 나눴다. 구사과는 뛰어난 코딩 실력을 이용해 각 칸 (r, c)에 있는 미세먼지의 양을 실시간으로 모니터링하는 시스템을 개발했다. (r, c)는 r행 c열을 의미한다.

공기청정기는 항상 1번 열에 설치되어 있고, 크기는 두 행을 차지한다. 공기청정기가 설치되어 있지 않은 칸에는 미세먼지가 있고, (r, c)에 있는 미세먼지의 양은 Ar,c이다.

1초 동안 아래 적힌 일이 순서대로 일어난다.

1. 미세먼지가 확산된다. 확산은 미세먼지가 있는 모든 칸에서 동시에 일어난다.
   • (r, c)에 있는 미세먼지는 인접한 네 방향으로 확산된다.
   • 인접한 방향에 공기청정기가 있거나, 칸이 없으면 그 방향으로는 확산이 일어나지 않는다.
   • 확산되는 양은 Ar,c/5이고 소수점은 버린다.
   • (r, c)에 남은 미세먼지의 양은 Ar,c - (Ar,c/5)×(확산된 방향의 개수) 이다.
2. 공기청정기가 작동한다.
   • 공기청정기에서는 바람이 나온다.
   • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
   • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
   • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

다음은 확산의 예시이다.

왼쪽과 오른쪽에 칸이 없기 때문에, 두 방향으로만 확산이 일어났다.

인접한 네 방향으로 모두 확산이 일어난다.

공기청정기가 있는 칸으로는 확산이 일어나지 않는다.

공기청정기의 바람은 다음과 같은 방향으로 순환한다.

방의 정보가 주어졌을 때, 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초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

---

시간에 따른 먼지의 변화를 잘 구현하면 되는 문제였다. 해야 할 일들이 이미 문제에 다 주어져있어서 그냥 구현만 하면 된다. 

 

문제를 풀면서 딱히 막히는 부분은 없었다. 하지만 공기청정기 작동이 구현하기 좀 귀찮았다. for문을 무려 8번을 써야 하다 보니 범위를 틀리지 않을까 조심해서 작성했다. 이것보다 좀 더 일반적으로 작동하는 코드를 작성할 수도 있지만 이 방법이 가장 정확할 것이라고 생각했다.

 

그리고 배열의 값이 5 미만이면 어차피 확산이 되지 않기 때문에 공기청정기를 나타내는 값인 -1부터 4까지는 별 다른 조치가 필요하지 않다고 생각했다. 하지만 그 부분을 바꾸면 단호하게 틀리더라. 도대체 이유가 뭘까....? 임의의 예시를 집어넣었을 때는 잘 되는 것 같았는데...

 

	#include <iostream>
	#include <cstring>
	using namespace std;
	const int dx[4] = { -1, 0, 1, 0 };
	const int dy[4] = { 0, 1, 0, -1 };
	int R, C, T;
	int room[51][51];
	int up_cleaner, down_cleaner;
	void diffuseDust() {
	int temp[51][51];
	memset(temp, 0, sizeof(temp));
	temp[up_cleaner][1] = -1;
	temp[down_cleaner][1] = -1;
	for (int r = 1; r <= R; r++) for (int c = 1; c <= C; c++) {
	//5 미만이면 어차피 확산안되는듯
	if (room[r][c] <= 0) continue;
	int cnt = 0;
	int spread = room[r][c] / 5;
	for (int dir = 0; dir < 4; dir++) {
	int nr = r + dx[dir], nc = c + dy[dir];
	if (nr < 1 || nr > R || nc < 1 || nc > C) continue;
	if (room[nr][nc] == -1) continue;
	temp[nr][nc] += spread;
	cnt++;
	}
	temp[r][c] += room[r][c] - spread * cnt;
	}
	memcpy(room, temp, sizeof(temp));
	}
	void activateCleaner() {
	//위쪽
	for (int r = up_cleaner - 1; r > 1; r--)
	room[r][1] = room[r - 1][1];
	for (int c = 1; c < C; c++)
	room[1][c] = room[1][c + 1];
	for (int r = 1; r < up_cleaner; r++)
	room[r][C] = room[r + 1][C];
	for (int c = C; c > 2; c--)
	room[up_cleaner][c] = room[up_cleaner][c - 1];
	room[up_cleaner][2] = 0;
	//아래쪽
	for (int r = down_cleaner + 1; r < R; r++)
	room[r][1] = room[r + 1][1];
	for (int c = 1; c < C; c++)
	room[R][c] = room[R][c + 1];
	for (int r = R; r > down_cleaner; r--)
	room[r][C] = room[r - 1][C];
	for (int c = C; c > 2; c--)
	room[down_cleaner][c] = room[down_cleaner][c - 1];
	room[down_cleaner][2] = 0;
	}
	int sumDust() {
	int ret = 0;
	for (int r = 1; r <= R; r++) for (int c = 1; c <= C; c++)
	ret += room[r][c];
	return ret;
	}
	int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);
	cin >> R >> C >> T;
	for (int r = 1; r <= R; r++) {
	for (int c = 1; c <= C; c++) {
	cin >> room[r][c];
	if (room[r][c] == -1) {
	if (!up_cleaner) up_cleaner = r;
	else down_cleaner = r;
	}
	}
	}
	while (T--) {
	diffuseDust();
	activateCleaner();
	}
	cout << sumDust() + 2 << "\n";
	return 0;
	}

view raw 17144.cpp hosted with ❤ by GitHub

 

후기)

오늘 삼성전자 인턴 서류를 냈다. 서류 통과하고 코테도 이런 문제만 나왔으면 좋겠다...ㅎㅎ