백준 7569번 토마토

백준 7569번 토마토

#include <iostream>
#include <string>
#include <vector>
#include<queue>
#include <algorithm>
using namespace std;
 
int H, M, N;
int Box[101][101][101];
queue<pair<pair<int, int>, int> > tmt;
int dy[6] = { 1,-1,0,0,0,0 };
int dx[6] = { 0,0,1,-1,0,0 };
int dz[6] = { 0,0,0,0,1,-1 };
int noTomato;
bool AllRipe(void) {
 
    for (int i = 0; i < H; i++)
        for (int j = 0; j < N; j++)
            for (int k = 0; k < M; k++)
                if (Box[j][k][i] == 0)
                    return 0;
    
    return true;
}
 
int Tomato(void) {
    if (tmt.empty())
        return -1;
    int day = 0;
    while (!tmt.empty()) {
        int currentSize = tmt.size();
        for (int i = 0; i < currentSize; i++) {
            int y = tmt.front().first.first;
            int x = tmt.front().first.second;
            int z = tmt.front().second;
 
            for (int i = 0; i < 6; i++) {
                int NextY = y + dy[i];
                int NextX = x + dx[i];
                int NextZ = z + dz[i];
                if (0 <= NextY && NextY < N && 0 <= NextX && NextX < M && 0 <= NextZ && NextZ < H) {
                    if (Box[NextY][NextX][NextZ] == 0) {
                        Box[NextY][NextX][NextZ] = 1;
                        tmt.push(make_pair(make_pair(NextY, NextX), NextZ));
                    }
                }
            }
            tmt.pop();
            if (tmt.empty()&&AllRipe())
                return day;
            else if (tmt.empty()&&!AllRipe()) {
                return -1;
            }
 
        }
        day++;
    }
 
}
int main() {
    ios_base::sync_with_stdio(0);
    cin.tie(0);
    cin >> M >> N >> H;
 
    for (int k = 0; k < H; k++) {
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                cin >> Box[i][j][k];
                if (Box[i][j][k] == 1) {
                    tmt.push(make_pair(make_pair(i, j), k));
                }
            }
        }
    }
    cout << Tomato();
 
    return 0;
}
 

3차원 배열이 너무 헷갈려서 중간에 삽질을 많이했네요. 

푸는 개념은 2차원 배열일 때 BFS문제와 동일합니다. 

currentsize를 하루가 지날때마다 갱신해주는 것에 주의합시다.