[BOJ] 17144번 미세먼지 안녕! c++

https://www.acmicpc.net/problem/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)×(확산된 방향의 개수) 이다.
  1. 공기청정기가 작동한다.
  • 공기청정기에서는 바람이 나온다.
  • 위쪽 공기청정기의 바람은 반시계방향으로 순환하고, 아래쪽 공기청정기의 바람은 시계방향으로 순환한다.
  • 바람이 불면 미세먼지가 바람의 방향대로 모두 한 칸씩 이동한다.
  • 공기청정기에서 부는 바람은 미세먼지가 없는 바람이고, 공기청정기로 들어간 미세먼지는 모두 정화된다.

다음은 확산의 예시이다.

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

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

공기청정기가 있는 칸으로는 확산이 일어나지 않는다.
공기청정기의 바람은 다음과 같은 방향으로 순환한다.

방의 정보가 주어졌을 때, T초가 지난 후 구사과의 방에 남아있는 미세먼지의 양을 구해보자.


입력 1
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

출력 1
188


입력 2
7 8 2
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

출력 2
188


입력 3
7 8 3
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

출력 3
186


입력 4
7 8 4
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

출력 4
178


입력 5
7 8 5
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

출력 5
172


입력 6
7 8 20
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

출력 6
71


입력 7
7 8 30
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

출력 7
52


입력 8
7 8 50
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

출력 8
46


#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstring>

using namespace std;

int R, C, T;
int map[50][50] = {0,};
int nmap[50][50] = {0,};
int tpx = -1, tpy = -1;
int dpx = -1, dpy = -1;
int air, nx, ny;

int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, -1, 1};

void clean() {
  for(int t = 0; t < T; t++) {
    memset(nmap, 0, sizeof(nmap));
    for(int i = 0; i < R; i++) {
      for(int j = 0; j < C; j++) {
        air = map[i][j];
        for(int k = 0; k < 4; k++) {
          if(map[i][j] > 0) {
            nx = i + dx[k];
            ny = j + dy[k];
            if(-1 < nx && nx < R && -1 < ny && ny < C && map[nx][ny] != -1) {
              nmap[nx][ny] += floor(air / 5);
              nmap[i][j] -= floor(air / 5);
            }
          }
        }
      }
    }

    for(int i = 0; i < R; i++) {
      for(int j = 0; j < C; j++) {
        map[i][j] += nmap[i][j];
        nmap[i][j] = map[i][j];
      }
    }

    for(int i = tpy + 1; i < C - 1; i++)
      nmap[tpx][i + 1] = map[tpx][i];
    for(int i = tpx; i > 0; i--)
      nmap[i - 1][C - 1] = map[i][C - 1];
    for(int i = C - 1; i > 0; i--)
      nmap[0][i - 1] = map[0][i];
    for(int i = 0; i < tpx; i++)
      nmap[i + 1][0] = map[i][0];

    for(int i = dpy + 1; i < C - 1; i++)
      nmap[dpx][i + 1] = map[dpx][i];
    for(int i = dpx; i < R - 1; i++)
      nmap[i + 1][C - 1] = map[i][C - 1];
    for(int i = C - 1; i > 0; i--)
      nmap[R - 1][i - 1] = map[R - 1][i];
    for(int i = R - 1; i > dpx; i--)
      nmap[i - 1][0] = map[i][0];

    memcpy(map, nmap, sizeof(map));
    map[tpx][tpy] = -1;
    map[tpx][tpy + 1] = 0;
    map[dpx][dpy] = -1;
    map[dpx][dpy + 1] = 0;
  }
}

int main() {
  int cnt = 0;

  scanf("%d %d %d", &R, &C, &T);
  for(int i = 0; i < R; i++) {
    for(int j = 0; j < C; j++) {
      scanf("%d", &map[i][j]);
      if(map[i][j] == -1 && tpx == -1 && tpy == -1) {
        tpx = i;
        tpy = j;
        dpx = i + 1;
        dpy = j;
      }
    }
  }
  clean();

  for(int i = 0; i < R; i++) {
    for(int j = 0; j < C; j++) {
      if(map[i][j] != -1)
        cnt += map[i][j];
    }
  }
  printf("%d\n", cnt);
}

좋은 웹페이지 즐겨찾기