[백준 17144] 미세먼지 안녕! C++

문제

백준 17144 미세먼지 안녕! C++

17144번: 미세먼지 안녕!

 

풀이

빡구현 문제입니다.

문제 이해 자체가 어렵진 않았지만, 구현과정에서 시간이 꽤 소요되었습니다.

index 에러를 주의하며 풀었습니다.

 

1. 입력을 받고 공기청정기의 위치를 저장합니다. (위 / 아래에 나눠서 저장)

2. 먼지를 확산시킵니다.

2-1. 해당칸에 먼지가 있다면 상하좌우로 확산시켜 temp 배열에 저장합니다. 확산할 수 없다면 확산하지 않습니다.

2-2. 먼지는 동시에 확산하므로 temp 배열에 저장시켜 놨다가, 한 번에 다시 map 배열에 더 합니다.

3. 공기청정기를 작동시킵니다.

3-1. 위 / 아래로 나눠 먼지를 이동시킵니다. 공지청정기 오른쪽은 무조건 0이 됩니다.

4. T 만큼 2~3 과정을 반복합니다.

5. 먼지 합을 출력합니다. 이 때 공기청정기는 계산하지 않습니다.

소스 코드

#include <iostream>
using namespace std;
 
int R, C, T;
int CleanUp = -1;
int CleanDown = -1;
int map[50][50];
int temp[50][50];
int ans = 0;
 
// 먼지 확산하기
void SpreadDust() {
    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            if (map[i][j] != -1 && map[i][j] != 0) {
                int dust = map[i][j] / 5;
                // 상
                if (i > 0 && map[i - 1][j] != -1) {
                    temp[i - 1][j] += dust;
                    map[i][j] -= dust;
                }
                // 하
                if (i + 1 < R && map[i + 1][j] != -1) {
                    temp[i + 1][j] += dust;
                    map[i][j] -= dust;
                }
                // 좌
                if (j > 0 && map[i][j - 1] != -1) {
                    temp[i][j - 1] += dust;
                    map[i][j] -= dust;
                }
                // 우
                if (j + 1 < C && map[i][j + 1] != -1) {
                    temp[i][j + 1] += dust;
                    map[i][j] -= dust;
                }
            }
        }
    }
    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            map[i][j] += temp[i][j];
            temp[i][j] = 0;
        }
    }
}
 
// 공기청정기 작동
void Purifier(int Up, int Down) {
    // 위쪽 반시계
    for (int i = Up - 1; i >= 1; i--) {
        map[i][0] = map[i - 1][0];
    }
    for (int i = 0; i < C - 1; i++) {
        map[0][i] = map[0][i + 1];
    }
    for (int i = 0; i < Up; i++) {
        map[i][C - 1] = map[i + 1][C - 1];
    }
    for (int i = C - 1; i >= 2; i--) {
        map[Up][i] = map[Up][i - 1];
    }
    map[Up][1] = 0;
    // 아래쪽 시계
    for (int i = Down + 1; i < R - 1; i++) {
        map[i][0] = map[i + 1][0];
    }
    for (int i = 0; i < C - 1; i++) {
        map[R - 1][i] = map[R - 1][i + 1];
    }
    for (int i = R - 1; i > Down; i--) {
        map[i][C - 1] = map[i - 1][C - 1];
    }
    for (int i = C - 1; i >= 2; i--) {
        map[Down][i] = map[Down][i - 1];
    }
    map[Down][1] = 0;
}
 
// 먼지 합 계산
void CalSum() {
    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            // 공기청정기가 아닌 경우에만 더함
            if (map[i][j] != -1) {
                ans += map[i][j];
            }
        }
    }
}
 
int main()
{
    // 입력
    cin >> R >> C >> T;
    for (int i = 0; i < R; i++) {
        for (int j = 0; j < C; j++) {
            cin >> map[i][j];
            // -1인 경우 청소기 위치 저장
            if (map[i][j] == -1) {
                if (CleanUp == -1) {
                    CleanUp = i;
                }
                else {
                    CleanDown = i;
                }
            }
        }
    }
 
    while (T--)
    {
        // 먼지 확산
        SpreadDust();
        // 공기청정기 작동
        Purifier(CleanUp, CleanDown);
    }
    CalSum();
    cout << ans;
    return 0;
}