Record my moment.

링크

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

문제

풀이

1. 토마토의 정보를 담을 배열 하나, 걸리는 시간을 담을 배열 하나해서 총 두개의 3차원 배열을 만들어준다.

2. 반복문을 통해 배열을 받아준다.

2-1. 입력을 보면, 한판을 다 받고 다음판을 받기 때문에 반복문을 배치할 때, H를 가장 밖에 둔다.

3. 배열에 있는 숫자가 1, 즉 익은 토마토일 때 시간을 나태내는 배열에 1을 넣으며, 더불어 queue에 넣고 bfs를 할 준비를 한다.

4. queue에 넣은 숫자들을 하나씩 꺼내고, 인접한 부분을 모두 탐색시킨다.

5. 3차원 배열이므로, h값 즉 zindex가 바뀌는 경우를 고려하여 추가한다.

6. 탐색을 진행하면서 안익은 토마토를 만났고, 해당 부분의 걸리는 시간이 0으로 아직 변경되지 않았을 때, 이동하기 전 토마토의 시간 + 1 값을 저장한다.

7. 탐색을 진행한 후, 다시 배열들의 탐색을 진행한다.

8. 토마토를 전부 익힐 수 있는지에 대한 여부를 판단하는 boolean 변수를 만들어 준다,.

9. 반복문을 통해 배열을 탐색하면서 토마토가 존재하는 경우(!= -1)인데 걸리는 시간이 0인 경우, 탐색이 진행되지 않은 것으로 판단하여,

설정한 boolean변수를 false로 바꾸고 break한다.

10. 그게 아니라면 계속 탐색을 진행해서 가장 큰 값을 정답으로 설정한다.

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.LinkedList;
import java.util.Queue;
import java.util.StringTokenizer;

public class Main {
    static int x = 0,y = 0,z = 0;
    static int[] dz = {-1,0,0,0,0,1};
    static int[] dy = {0,1,0,-1,0,0};
    static int[] dx = {0,0,1,0,-1,0};

    static Queue<Integer[]> queue = new LinkedList<>();
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());

        x = Integer.parseInt(st.nextToken());
        y = Integer.parseInt(st.nextToken());
        z = Integer.parseInt(st.nextToken());

        // 입력 index
        int[][][] arr = new int[z][y][x];
        int[][][] dist = new int[z][y][x];

        for(int i = 0; i<z; i++){
            for(int j = 0; j<y; j++){
                st = new StringTokenizer(br.readLine());
                for(int k = 0; k<x; k++){
                    int num = Integer.parseInt(st.nextToken());
                    arr[i][j][k] = num;
                    if(num == 1){
                        dist[i][j][k] = 1;
                        queue.offer(new Integer[]{i,j,k});
                    }
                }
            }
        }
        int ans = 0;

        while(!queue.isEmpty()){
            Integer[] tmp = queue.poll();

            int tmpz = tmp[0];
            int tmpy = tmp[1];
            int tmpx = tmp[2];

            for(int i = 0; i<dz.length; i++){
                int nz = tmpz + dz[i];
                int ny = tmpy + dy[i];
                int nx = tmpx + dx[i];

                if(nz < 0 || ny < 0 || nx <0 || nz >= z || ny >= y || nx >= x){
                    continue;
                }
                if(arr[nz][ny][nx] == 0 && dist[nz][ny][nx] == 0){
                    dist[nz][ny][nx] = dist[tmpz][tmpy][tmpx] + 1;
                    queue.offer(new Integer[] {nz,ny,nx});
                }
            }
        }

        boolean possible = true;

        for(int i = 0; i<z; i++){
            if (!possible) {
                ans = 0;
                break;
            }
            for(int j = 0;  j<y; j++){
                for(int k = 0; k<x; k++){
                    if (!possible) {
                        ans = 0;
                        break;
                    }
                    if(arr[i][j][k] != -1&&dist[i][j][k] == 0) {
                        possible = false;
                        if (!possible) {
                            ans = 0;
                            break;
                        }
                    }
                    ans = Math.max(ans, dist[i][j][k]);
                }
            }
        }
        System.out.println(ans-1);
    }

    }

후기

3차원 배열 탐색할 때, 처음에 zindex를 -1로 옮기고 xy도 같이 옮겨버려서 답이 안나왔음.

처음에 bfs 메소드로 만들고 싶어서 static 선언했는데, 중간에 수업 듣느라 까먹어서 dx dy dz static 해놓은 것도 까먹음

다음번엔 메소드 단위로 짤라서 만들어보고 싶음

다음엔 주석을 좀 써볼까 고민 중