[백준] 7569: 토마토 (3차원 그래프탐색, bfs) - JAVA

알고리즘 분류 (그래프 탐색)

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

 

 

 

# 문제

 

# 필요개념

지금까지 했던 2차원 배열 탐색과 달리, 토마토가 들어있는 3차원 상자를 탐색해야 한다. 그래서 삼차원 배열을 생성해주었다. 지금까지 BFS로 그래프탐색을 할 때 방문 여부를 확인하는 isVisit 배열을 생성했었는데, 이 문제는 애초에 전체 토마토를 익게 할 수 있는 "시간" 을 물어보고 있기 때문에, 시간을 넣어줄 배열(map)을 생성해주었다. 토마토 입력이 0인 자리에 초기값을 -1로 설정해주어서 -1이면 방문하지 않은 곳임을 알 수 있도록 했다.

 

모든 익은 토마토들은 동시에 주변을 익게 만들기 때문에, 입력이 1인 토마토들을 일단 모두 Queue에 담고, 다 담은 후에 bfs 탐색을 시작하도록 했다.

 

탐색을 마친 후에는 map을 돌면서 -1인 곳이 있으면 방문하지 못한 노드가 있다는 의미이므로 -1을 출력, 없다면 map의 값들 중 가장 큰 값을 출력하도록 했다. 값이 가장 크다는 것은 익기까지 가장 오랜시간이 걸렸다는 의미 즉 마지막으로 익은 토마토라는 의미이다.

 

# Code

import java.io.*;
import java.util.*;

import static java.lang.Math.*;

public class Main {
    static int M;
    static int N;
    static int H;
    static int[] dz = {1, -1, 0, 0, 0, 0}; // 위, 아래
    static int[] dy = {0, 0, -1, 1, 0, 0}; // 북, 남
    static int[] dx = {0, 0, 0, 0, -1, 1}; // 서, 동
    static int[][][] tomatoes;
    static int[][][] map;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int ans = 0;
        M = Integer.parseInt(st.nextToken());
        N = Integer.parseInt(st.nextToken());
        H = Integer.parseInt(st.nextToken());
        tomatoes = new int[H][N][M];
        map = new int[H][N][M];

        for (int i = 0 ; i < H ; i++) {
            for (int j = 0 ; j < N ; j++) {
                st = new StringTokenizer(br.readLine());
                for (int k = 0 ; k < M ; k++) {
                    tomatoes[i][j][k] = Integer.parseInt(st.nextToken());
                    if (tomatoes[i][j][k] == 0) map[i][j][k] = -1;
                }
            }
        }

        Queue<int[]> queue = new LinkedList<>();
        for (int z = 0; z < H; z++) {
            for (int y = 0; y < N; y++) {
                for (int x = 0; x < M; x++) {
                    if (tomatoes[z][y][x] == 1) {
                        queue.add(new int[]{z, y, x});
                    }
                }
            }
        }
        bfs(queue);

        for (int z = 0; z < H; z++) {
            for (int y = 0; y < N; y++) {
                for (int x = 0; x < M; x++) {
                    if (map[z][y][x] == -1) {
                        System.out.println(-1);
                        return;
                    }
                    ans = Math.max(ans, map[z][y][x]);
                }
            }
        }
        System.out.println(ans);
    }

    private static void bfs(Queue<int[]> queue) {
        while (!queue.isEmpty()) {
            int[] now = queue.poll();
            for (int i = 0 ; i < dy.length ; i++) {
                int nowZ = now[0] + dz[i];
                int nowY = now[1] + dy[i];
                int nowX = now[2] + dx[i];

                if (canExplore(nowZ, nowY, nowX)) {
                    queue.add(new int[]{nowZ, nowY, nowX});
                    map[nowZ][nowY][nowX] = map[now[0]][now[1]][now[2]] + 1;
                }
            }
        }
    }

    private static boolean canExplore(int z, int y, int x) {
        return z >= 0 && z < H && y >= 0 && y < N && x >= 0 && x < M && map[z][y][x] == -1;
    }
}

 

 

# 결과