https://www.acmicpc.net/problem/2667
처음에 많이 헤맸던 문제였는데 다시 풀었을 때도 헤매네 ㅎ
출력 : 총 단지 수 , 단지내 집의 수 오름차순 정렬
정렬하기 위해 ArrayList를 활용하였고 탐색하면서 방문할 때마다 집의 수를 카운트해줬다. 탐색이 끝나면 ArrayList에 값을 넣어주었고 마지막으로 정렬해서 결과값을 출력하였다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.Queue;
public class Main {
static int n;
static int[][] map;
static boolean[][] check;
static Queue<Node> queue;
static int[] dx = {1, 0, -1, 0}, dy = {0, 1, 0, -1};
static ArrayList<Integer> list;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
n = Integer.parseInt(br.readLine());
map = new int[n][n];
check = new boolean[n][n];
list = new ArrayList<>(); // 단지
// 입력받기
for (int i = 0; i < n; i++) {
String input = br.readLine();
for (int j = 0; j < n; j++) {
map[i][j] = input.charAt(j) - '0';
}
}
// 탐색
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
if (map[i][j] == 1 && !check[i][j]) {
bfs(i, j);
}
}
}
Collections.sort(list);
System.out.println(list.size());
for (int a : list) {
System.out.println(a);
}
}
public static void bfs(int x, int y) {
queue = new LinkedList<>();
int cnt = 0; // 단지내 집의 수
queue.add(new Node(x, y));
check[x][y] = true;
cnt++; // 방문할 때마다 +1 증가
while (!queue.isEmpty()) {
Node node = queue.poll();
for (int i = 0; i < 4; i++) {
int nx = node.x + dx[i];
int ny = node.y + dy[i];
if (nx >= 0 && ny >= 0 && nx < n && ny < n) {
if (map[nx][ny] == 1 & !check[nx][ny]) {
queue.add(new Node(nx, ny));
check[nx][ny] = true;
cnt++;
}
}
}
}
list.add(cnt); // 탐색 종료하면 단지에 추가
}
}
class Node {
int x;
int y;
public Node(int x, int y) {
this.x = x;
this.y = y;
}
}