[JAVA-자바] 2638번: 치즈

 

소스코드:

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

public class Main {

  public static void main(String[] args) throws IOException {
    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    StringTokenizer st = new StringTokenizer(br.readLine());
    int height = Integer.parseInt(st.nextToken());
    int width = Integer.parseInt(st.nextToken());

    int[][] board = new int[height][width];
    int total = 0;
    for (int y = 0; y < height; y++) {
      st = new StringTokenizer(br.readLine());
      for (int x = 0; x < width; x++) {
        board[y][x] = Integer.parseInt(st.nextToken());
        if (board[y][x] == 1) {
          total++;
        }
      }
    }

    int time = 0;
    int[] dx = {0, 0, -1, 1};
    int[] dy = {-1, 1, 0, 0};

    while (total > 0) {
      boolean[][] visited = new boolean[height][width];
      int[][] cheese = new int[height][width];

      Queue<int[]> queue = new LinkedList<>();
      queue.add(new int[]{0, 0});

      while (!queue.isEmpty()) {
        int[] current = queue.poll();
        int x = current[0];
        int y = current[1];

        for (int i = 0; i < 4; i++) {
          int nextX = x + dx[i];
          int nextY = y + dy[i];

          if (nextX < 0 || nextX >= width || nextY < 0 || nextY >= height) {
            continue;
          }

          if (visited[nextY][nextX]) {
            continue;
          }

          if (board[nextY][nextX] == 0) {
            // air
            visited[nextY][nextX] = true;
            queue.add(new int[]{nextX, nextY});
          } else if (board[nextY][nextX] == 1) {
            // cheese
            cheese[nextY][nextX]++;
          }
        }
      }

      // del
      for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
          if (cheese[y][x] >= 2) {
            board[y][x] = 0;
            total--;
          }
        }
      }
      time++;
    }
    bw.write(String.valueOf(time));
    bw.close();
    br.close();
  }
}

 

글의 내용 중 잘못된 점이나 수정이 필요한 부분, 혹은 궁금한 사항이 있다면 언제든 댓글로 남겨주시면 감사하겠습니다.

여러분의 피드백은 더 나은 글을 작성하는 데 큰 도움이 됩니다. 감사합니다.