[BOJ] 2636번 치즈 자바

문제

문제

아래 <그림 1>과 같이 정사각형 칸들로 이루어진 사각형 모양의 판이 있고, 그 위에 얇은 치즈(회색으로 표시된 부분)가 놓여 있다. 판의 가장자리(<그림 1>에서 네모 칸에 X친 부분)에는 치즈가 놓여 있지 않으며 치즈에는 하나 이상의 구멍이 있을 수 있다.

이 치즈를 공기 중에 놓으면 녹게 되는데 공기와 접촉된 칸은 한 시간이 지나면 녹아 없어진다. 치즈의 구멍 속에는 공기가 없지만 구멍을 둘러싼 치즈가 녹아서 구멍이 열리면 구멍 속으로 공기가 들어가게 된다. <그림 1>의 경우, 치즈의 구멍을 둘러싼 치즈는 녹지 않고 ‘c’로 표시된 부분만 한 시간 후에 녹아 없어져서 <그림 2>와 같이 된다.

<그림 1> 원래 치즈 모양

다시 한 시간 후에는 <그림 2>에서 ‘c’로 표시된 부분이 녹아 없어져서 <그림 3>과 같이 된다.

<그림 2> 한 시간 후의 치즈 모양

<그림 3> 두 시간 후의 치즈 모양

<그림 3>은 원래 치즈의 두 시간 후 모양을 나타내고 있으며, 남은 조각들은 한 시간이 더 지나면 모두 녹아 없어진다. 그러므로 처음 치즈가 모두 녹아 없어지는 데는 세 시간이 걸린다. <그림 3>과 같이 치즈가 녹는 과정에서 여러 조각으로 나누어 질 수도 있다.

입력으로 사각형 모양의 판의 크기와 한 조각의 치즈가 판 위에 주어졌을 때, 공기 중에서 치즈가 모두 녹아 없어지는 데 걸리는 시간과 모두 녹기 한 시간 전에 남아있는 치즈조각이 놓여 있는 칸의 개수를 구하는 프로그램을 작성하시오.

풀이

#bfs

bfs를 이용하여 녹을 치즈와 녹지 않을 치즈를 구분해낸다. bfs 탐색을 진행하며 치즈가 외부 공기와 맞닿아 있는 경우 배열에 표시를 해주고 빈 공간일 경우에 큐에 집어넣는다. 이때 빈 공간이어도 치즈에 둘러싸여 있으면 공기가 유입되지 않는데, bfs탐색을 진행하면서 치즈에 사방으로 둘러싸여 있는 경우 탐색되지 않기 때문에 신경쓰지 않아도 된다.

 

탐색을 치즈가 아닌 빈 공간을 기준으로 한 점은 내부 빈 공간과 외부 빈 공간을 구분하기 위해서이다.

 

메인 함수에서는 무한루프를 돌며 남은 치즈 개수를 계산한다. 만약 치즈가 다 녹았다면 무한루프를 빠져나온다.

이때 주의해야 할 점이 만일 치즈를 모두 녹이는 데에 시간이 1이 소요되거나 치즈가 처음부터 없는 경우 예외처리를 해주어야 한다.

 

코드

package BOJ.bfs;

import java.util.*;
import java.io.*;
public class No2636_치즈 {
    static int[][] arr;
    static int H, W;
    static int[] dx={-1,1,0,0};
    static int[] dy={0,0,-1,1};
    static boolean[][] visited;
    static int time=0;
    static int remain=0;
    public static void main(String[] args) throws IOException{
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st=new StringTokenizer(br.readLine());

        H=Integer.parseInt(st.nextToken());
        W= Integer.parseInt(st.nextToken());
        arr=new int[H][W];
        visited=new boolean[H][W];

        for(int i=0;i<H;i++){
            st=new StringTokenizer(br.readLine());
            for(int j=0;j<W;j++){
                arr[i][j]=Integer.parseInt(st.nextToken());
            }
        }

        int firstCheese=calcCount();
        int count=0;
        while(true){
            time++;
            bfs(0,0);
            visited=new boolean[H][W]; // visited 초기화

            remain=count;
            count=calcCount(); //치즈개수 계산
            if(count==0) break; //다 녹이면 break
        }

        System.out.println(time);
        System.out.println(time==1? firstCheese:remain);
    }

    static void bfs(int x, int y){
        Queue<Node> q=new LinkedList<>();
        q.offer(new Node(x,y));
        visited[x][y]=true;

        while(!q.isEmpty()){
            Node cur=q.poll();
            int px=cur.x;
            int py=cur.y;

            for(int i=0;i<4;i++){
                int nx=px+dx[i];
                int ny=py+dy[i];

                if(nx<0||nx>=H||ny<0||ny>=W||visited[nx][ny]) continue;

                if(arr[nx][ny]==1){
                    arr[nx][ny]=2;
                    visited[nx][ny]=true;
                }
                if(arr[nx][ny]==0){
                    q.offer(new Node(nx,ny));
                    visited[nx][ny]=true;
                }
            }
        }

        // 치즈 녹이기
        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(arr[i][j]==2) arr[i][j]=0;
            }
        }
    }

    //개수 계산
    static int calcCount(){
        int count=0;

        for(int i=0;i<H;i++){
            for(int j=0;j<W;j++){
                if(arr[i][j]==1) count++;
            }
        }

        return count;
    }

    static class Node {
        int x, y;
        public Node(int x, int y){
            this.x=x;
            this.y=y;
        }
    }
}