[백준] 17144번 : 미세먼지 안녕! (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초가 지난 후 구사과 방에 남아있는 미세먼지의 양을 출력한다.

 

 

 

 

 

 

풀이)

 

// 17144. 미세먼지 안녕!
#include <iostream>
#include <vector>
 
using namespace std;
 
int n, m, t;
vector<vector<int>> vec;
pair<int, int> air;
 
int dx[] = { -1, 0, 1, 0, 0 };
int dy[] = { 0, -1, 0, 1, 0 };
 
void spread()
{
    vector<vector<int>> temp;
    temp.resize(n, vector<int>(m, 0));
 
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            int total = 0;
            if (vec[i][j] > 0)
            {
                for (int k = 0; k < 5; k++)
                {
                    int nx = i + dx[k];
                    int ny = j + dy[k];
 
                    if (k == 4)
                    {
                        temp[nx][ny] += (vec[i][j] - total);
                        continue;
                    }
 
                    if (0 > nx || nx >= n) continue;
                    if (0 > ny || ny >= m) continue;
                    if (vec[nx][ny] == -1) continue;
 
                    total += vec[i][j] / 5;
                    temp[nx][ny] += (vec[i][j] / 5);
                }
            }
        }
    }
    vec[air.first][air.second] = -1;
    vec[air.first - 1][air.second] = -1;
    vec = temp;
}
 
void air_cleaner(int x, int y)
{
    vec[air.first][air.second] = 0;
    vec[air.first - 1][air.second] = 0;
    int tx = x, ty = y;
    
    int nx = -1;
    int ny = 0;
 
    while (1)
    {
        tx += nx; ty += ny;
        if (tx + nx < 0)
        {
            nx = 0;
            ny = 1;
        }
        else if (ty + ny >= m)
        {
            nx = 1;
            ny = 0;
        }
        else if (tx + nx > x)
        {
            nx = 0;
            ny = -1;
        }
        vec[tx][ty] = vec[tx + nx][ty + ny];
        if (tx + nx == x && ty + ny == y) break;
    }
    vec[air.first][air.second] = 0, vec[air.first - 1][air.second] = 0;
 
    tx = x + 1, ty = y;
    nx = 1, ny = 0;
    while (1)
    {
        tx += nx; ty += ny;
        if (tx + nx >= n)
        {
            nx = 0;
            ny = 1;
        }
        else if (ty + ny >= m)
        {
            nx = -1;
            ny = 0;
        }
        else if (tx + nx <= x)
        {
            nx = 0;
            ny = -1;
        }
 
        vec[tx][ty] = vec[tx + nx][ty + ny];
        if (tx + nx == x + 1 && ty + ny == y) break;
    }
    vec[air.first][air.second] = 0, vec[air.first - 1][air.second] = 0;
}
 
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
 
    cin >> n >> m >> t;
    vec.resize(n, vector<int>(m, 0));
 
    int a;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            cin >> a;
            vec[i][j] = a;
            if (a == -1)
            {
                air.first = i, air.second = j;
            }
 
        }
    }
 
    while (t--)
    {
        spread();
        air_cleaner(air.first - 1, air.second);
    }
 
    int ans = 0;
    for (auto i : vec) for (auto j : i)
    {
        if (j > 0) ans += j;
    }
 
    cout << ans;
 
    return 0;
}
 

---

출처 : https://www.acmicpc.net/problem/17144 

17144번: 미세먼지 안녕!