백준 7569 토마토(c++)

https://www.acmicpc.net/problem/7569

7569번: 토마토

기본적인 BFS문제와 유사한데, 높이가 주어져서 2차원이 아닌 3차원으로 표현해야 합니다.

더불어 이동 방향도 상, 하, 좌, 우에서 앞, 뒤가 추가된 6가지로 표현할 수 있습니다.

이동 방향이 행, 열, 높이 총 3개가 있으므로 원소를 3개 참조해야 합니다. 이러한 경우에 c++에서 제공하는 tuple 헤더를 이용해서 해당 원소를 한 번에 참조할 수 있게 되고 굉장히 편리합니다. pair를 2번 써도 상관없습니다. 그런데 코드의 가독성 차원에서 개인적으로 저는 tuple을 쓰는 게 더 좋은 것 같습니다.

<코드>

#include <iostream>
#include <queue>
#include <tuple>
#include <algorithm>
using namespace std;
int box[101][101][101];
int mov[6][3] = {{1,0,0},{0,1,0},{0,0,1},{-1,0,0},{0,-1,0},{0,0,-1}};
bool visited[101][101][101] = {false};
queue<tuple<int,int,int>> q;
int m,n,h;
int ans = 0;

void bfs() 
{
    while(!q.empty())
    {
        int a, b, c;
        tuple<int,int,int> t = q.front();
        a = get<0>(t);
        b = get<1>(t);
        c = get<2>(t);
        q.pop();
        for(int i=0;i<6;i++)
        {    
            int x = a + mov[i][0];
            int y = b + mov[i][1];
            int z = c + mov[i][2]; 
            if(x>=0 && y>=0 && z>=0 && x<h && y<n && z<m) 
            {
                if(box[x][y][z] == 0 && !visited[x][y][z])
                {
                    visited[x][y][z] = true;
                    q.push(make_tuple(x,y,z));
                    box[x][y][z] = box[a][b][c] + 1;
                }
            }

        }
    }
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);
    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++)
            {
                cin>>box[i][j][k]; // 높이, 열, 행
                if(box[i][j][k]==1) 
                { 
                    q.push(make_tuple(i,j,k));
                }
            }
        }
    }

    bfs();

    for(int i=0;i<h;i++)
    {
        for(int j=0;j<n;j++) 
        {
            for(int k=0;k<m;k++)
            {
                if(box[i][j][k]==0)
                { 
                    cout<<-1<<'\n';
                    return 0;
                }
                ans = max(ans, box[i][j][k]);
            }
        }
    }
   
    cout<<ans-1<<'\n';
    return 0;


    
}

제일 마지막에 ans-1을 한 이유는 익지 않은 토마토가 없는 경우에도 box의 최댓값은 1이기 때문에 -1을 해준 것 입니다.