[백준 7569] 토마토(3차원)

백준 7569 토마토(3차원)

이 문제는 토마토(2차원)이랑 높이가 생겼다는 것 빼곤 똑같다. 2차원에서 위, 아래(높이 방향)이 추가 되었으므로 체크해야하는 위치가 2가지가 늘었다. 하지만 푸는 방식은 토마토(2차원)이랑 같다. (토마토(2차원) 풀이 보러가기)

#include<cstdio>
#include<queue>
 
using namespace std;
 
queue<pair<pair<int, int>, int>> q;
int map[101][101][101] = { 0 };
bool visit[101][101][101] = { false };
int dx[] = { -1,1,0,0,0,0 };
int dy[] = { 0,0,0,0,-1,1};
int dz[] = { 0,0,-1,1,0,0 };
 
int main()
{
    int m, n, h;
    scanf("%d %d %d", &m, &n, &h);
    for (int k = 0; k < h; k++)
    {
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < m; j++)
            {
                scanf("%d", &map[i][j][k]);
                if (map[i][j][k] == 1)
                {
                    q.push(make_pair(make_pair(i, j), k));
                    visit[i][j][k] = true;
                }
                else if (map[i][j][k] == -1)
                {
                    visit[i][j][k] = true;
                }
            }
        }
    }
 
    int ans = -1;
 
    while (!q.empty())
    {
        int size = q.size();
        ++ans;
        while (size--)
        {
            pair<pair<int, int>, int> now = q.front(); q.pop();
 
            for (int i = 0; i < 6; i++)
            {
                int qx, qy, qz;
                qx = now.first.second  + dx[i];
                qy = now.first.first + dy[i];
                qz = now.second + dz[i];
 
                if (qx < 0 || qx >= m || qy < 0 || qy >= n || qz < 0 || qz >= h)continue;
                if (!visit[qy][qx][qz] && map[qy][qx][qz] == 0)
                {
                    visit[qy][qx][qz] = true;
                    q.push(make_pair(make_pair(qy, qx), qz));
                }
            }
        }
    }
 
    for (int k = 0; k < h; k++)
    {
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < m; j++)
            {
                if (!visit[i][j][k])return !printf("-1");
            }
        }
    }
 
    printf("%d", ans);
    return 0;
}