백준 2573번 빙산 [ Java ]

2573번: 빙산

기존에 BFS 기본 코드에서 새로운 melt라는 새로운 배열을 추가해 빙산의 영역기준으로 네 방향에 값들이 0보다 작거나 같으면 이후에 빼주는 코드를 작성했습니다.

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

public class Main {

    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int[] dx = new int[]{1, 0, -1, 0};
        int[] dy = new int[]{0, 1, 0, -1};
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());
        int[][] field = new int[N][M];
        for (int i=0; i<N; i++)
            field[i] = Arrays.stream(br.readLine().split(" "))
                        .mapToInt(Integer::parseInt)
                        .toArray();
        int year = 0;
        do {
            int[][] melt = new int[N][M];
            boolean[][] visited = new boolean[N][M];
            int ice = 0;
            for (int i=0; i<N; i++)
                for (int j=0; j<M; j++){
                    if (visited[i][j] || field[i][j]<=0) continue;
                    ice++;
                    Queue<int[]> Q = new LinkedList<>();

                    Q.offer(new int[]{i, j});
                    visited[i][j] = true;
                    while (!Q.isEmpty()){
                        int[] cur = Q.poll();
                        for (int dir=0; dir<4; dir++){
                            int nx = cur[1] + dx[dir];
                            int ny = cur[0] + dy[dir];
                            if (nx<0 || nx>=M || ny<0 || ny>=N) continue;
                            if (visited[ny][nx]) continue;
                            if (field[ny][nx]<=0){
                                melt[cur[0]][cur[1]]--;
                                continue;
                            }
                            visited[ny][nx] = true;
                            Q.offer(new int[]{ny, nx});
                        }
                    }
                }
            if (ice==0){
                System.out.print(0);
                System.exit(0);
            }
            if (ice>=2){
                System.out.print(year);
                System.exit(0);
            }
            for (int i=0; i<N; i++)
                for (int j=0; j<M; j++)
                    field[i][j]+=melt[i][j];

        }while(++year>0);
    }
}