[2667/JAVA] 단지번호붙이기

• 입력: 가로세로 크기 N, N*N 단지 정보(1: 아파트 있음 0: 없음)
  
• 출력: 단지의 개수와 오름차순으로 각 단지의 아파트 개수 출력
• 알고리즘: BFS
• 소스코드
import java.util.*; class pair{ int x, y; pair(int x, int y){ this.x = x; this.y = y; } } class Ascending implements Comparator<Integer>{ @Override public int compare(Integer o1, Integer o2){ return o1.compareTo(o2); } } public class Main { static int[][] complex; static int[][] mov = new int[][] {{-1, 0},{0, -1},{1, 0},{0, 1}}; static int bfs(int n, int x, int y){ Queue myQ = new LinkedList(); myQ.add(new pair(x, y)); pair temp; int total = 0; while(!myQ.isEmpty()){ temp = (pair)myQ.poll(); if(complex[temp.x][temp.y] == 0) continue; total++; complex[temp.x][temp.y] = 0; for(int i=0; i<4; i++){ int mX = temp.x + mov[i][0], mY = temp.y+ mov[i][1]; if(mX < 0 || mX >= n || mY < 0 || mY >= n) continue; if(complex[mX][mY] != 1) continue; myQ.add(new pair(mX, mY)); } } return total; } public static void main(String[] args) { Scanner sc = new Scanner(System.in); int n; List<Integer> arr = new ArrayList<Integer>(); n = sc.nextInt(); complex = new int[n][n]; String input = ""; sc.nextLine(); for(int i=0; i<n; i++){ input = sc.nextLine(); for(int j=0; j<n; j++){ complex[i][j] = input.charAt(j) - '0'; } } int total = 0; for(int i=0; i<n; i++){ for(int j=0; j<n; j++){ if(complex[i][j] == 1){ total++; arr.add(bfs(n, i, j)); } } } System.out.println(total); Ascending ascending = new Ascending(); Collections.sort(arr, ascending); for(Integer element : arr){ System.out.println(element); } } }