토마토 7569

문제

접근 방법

3차원 배열을 이용하여서 풀어주면 된다. BFS를 사용해야 할 것이다. 왜냐하면 서서히 퍼져나가는 느낌으로 풀어줘야 하는 것이기에 DFS를 사용하면 다른 답이 나올 것이다.

코드

#include <iostream>
#include <queue>
#include <algorithm>
using namespace std;
int map[101][101][101];
int dx[] = {-1, 1, 0, 0, 0, 0}, dy[] = {0, 0, -1, 1, 0, 0}, dz[] = {0, 0, 0, 0, -1, 1};
struct node
{
    int z, y, x;
};
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    queue<node> bfs;
    int M, N, H, count = 0, day = 0;
    cin >> M >> N >> H;
    for (int i = 0; i < H; ++i)
    {
        for (int j = 0; j < N; ++j)
        {
            for (int k = 0; k < M; ++k)
            {
                int n;
                cin >> n;
                if (n == 1)
                    bfs.push({i, j, k});
                else if (n == 0)
                    ++count;
                map[i][j][k] = n;
            }
        }
    }
    if (count == 0)
    {
        cout << '0';
        return 0;
    }
    while (!bfs.empty())
    {
        node n = bfs.front();
        bfs.pop();
        int nextDay = map[n.z][n.y][n.x] + 1;
        for (int i = 0; i < 6; ++i)
        {
            node next = {n.z + dz[i], n.y + dy[i], n.x + dx[i]};
            if (next.z >= 0 && next.z < H && next.y >= 0 && next.y < N && next.x >= 0 && next.x < M && !map[next.z][next.y][next.x])
            {
                map[next.z][next.y][next.x] = nextDay;
                bfs.push(next);
                day = max(day, nextDay);
                --count;
            }
        }
    }
    if (count > 0)
    {
        cout << "-1";
    }
    else
    {
        cout << day - 1;
    }
    return 0;
}

풀이

모두 익었는지는 int 변수 하나를 두어서 확인해주면 된다.
이미 익은 토마토가 1부터 시작하기에 day에 1만큼 빼줘야 한다.