백준 17144 미세먼지 안녕! 풀이 (c++)

문제

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

 

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

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

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

 

#include <iostream>
#include <vector>

using namespace std;

int R, C, T, iResult = 0;
int iRoom[51][51] = { 0 };
const int dy[] = { -1, 0, 1, 0 };
const int dx[] = { 0, 1, 0, -1 };
vector<pair<int, int>> vCleaner;

void Explosive()
{
    int temp[51][51] = { 0 };

    for (int y = 0; y < R; ++y)
    {
        for (int x = 0; x < C; ++x)
        {
            if (iRoom[y][x] > 0)
            {
                int amount = iRoom[y][x] / 5;
                int count = 0;

                for (int i = 0; i < 4; ++i)
                {
                    int ny = y + dy[i];
                    int nx = x + dx[i];
                    if (ny < 0 || nx < 0 || ny >= R || nx >= C || iRoom[ny][nx] == -1)
                        continue;

                    temp[ny][nx] += amount;
                    count++;
                }
                temp[y][x] += iRoom[y][x] - (amount * count);
            }
        }
    }

    for (int i = 0; i < R; ++i)
    {
        for (int j = 0; j < C; ++j)
        {
            if (iRoom[i][j] != -1)
                iRoom[i][j] = temp[i][j];
        }
    }
}

void AirClean()
{
    int iTop = vCleaner[0].first;
    int iBottom = vCleaner[1].first;

    for (int i = iTop - 1; i > 0; --i)
        iRoom[i][0] = iRoom[i - 1][0];
    for (int i = 0; i < C - 1; ++i) 
        iRoom[0][i] = iRoom[0][i + 1];
    for (int i = 0; i < iTop; ++i)
        iRoom[i][C - 1] = iRoom[i + 1][C - 1];
    for (int i = C - 1; i > 1; --i)
        iRoom[iTop][i] = iRoom[iTop][i - 1];
    iRoom[iTop][1] = 0;

    for (int i = iBottom + 1; i < R - 1; ++i) 
        iRoom[i][0] = iRoom[i + 1][0];
    for (int i = 0; i < C - 1; ++i)
        iRoom[R - 1][i] = iRoom[R - 1][i + 1];
    for (int i = R - 1; i > iBottom; --i) 
        iRoom[i][C - 1] = iRoom[i - 1][C - 1];
    for (int i = C - 1; i > 1; --i) 
        iRoom[iBottom][i] = iRoom[iBottom][i - 1];
    iRoom[iBottom][1] = 0;
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL); cout.tie(NULL);
    cin >> R >> C >> T;

    for (int i = 0; i < R; ++i)
    {
        for (int j = 0; j < C; ++j)
        {
            cin >> iRoom[i][j];
            if (iRoom[i][j] == -1)
                vCleaner.push_back({ i, j });
        }
    }

    for (int i = 0; i < T; ++i)
    {
        Explosive();
        AirClean();
    }

    for (int i = 0; i < R; ++i)
    {
        for (int j = 0; j < C; ++j)
        {
            if (iRoom[i][j] > 0)
                iResult += iRoom[i][j];
        }
    }

    cout << iResult;
    return 0;
}

 

코드량이 길어져서 복잡해 지는 문제였다. 어차피1열에만 공기청정기가 있는점을 망각하면 햇갈리기 쉽다.