[JAVA-자바] 2667번: 단지번호붙이기

 

소스코드:

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class Main {
    static BufferedReader br;
    static BufferedWriter bw;

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

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

        int width = Integer.parseInt(br.readLine());
        int height = width;

        boolean[][] visited = new boolean[height + 2][width + 2];
        boolean[][] board = new boolean[height + 2][width + 2];

        for (int y = 1; y <= height; y++) {
            String line = br.readLine();
            for (int x = 1; x <= width; x++) {
                board[y][x] = line.charAt(x - 1) == '1';
            }
        }
        br.close();

        List<Integer> list = new ArrayList<>();

        for (int y = 1; y <= height; y++) {
            for (int x = 1; x <= width; x++) {
                if (!board[y][x] || visited[y][x]) {
                    continue;
                }
                // solve
                int total = 0;
                Queue<int[]> queue = new LinkedList<>();
                queue.add(new int[]{x, y});
                visited[y][x] = true;

                while (!queue.isEmpty()) {
                    int[] current = queue.poll();
                    int xx = current[0];
                    int yy = current[1];
                    total++;

                    for (int i = 0; i < 4; i++) {
                        int calcX = xx + dx[i];
                        int calcY = yy + dy[i];
                        if (!visited[calcY][calcX] && board[calcY][calcX]) {
                            visited[calcY][calcX] = true;
                            queue.add(new int[]{calcX, calcY});
                        }
                    }
                }
                list.add(total);
            }
        }

        Collections.sort(list);
        bw.write(list.size() + "\n");
        for (Integer item : list) {
            bw.write(item + "\n");
        }
        bw.close();
    }
}

 

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

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