import java.io.BufferedReader; import java.io.InputStreamReader; import java.io.FileReader; import java.io.IOException; import java.util.LinkedList; import java.util.Stack; import java.util.StringTokenizer; import java.util.ArrayList; import java.util.Collections; public class Main { static int N; //static int[][] map; // 전역 변수 사용하기 vs 함수 매개변수로 계속 전달하기 vs 함수 내부 함수로 정의하기 public static void main(String[] args) throws Exception { BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); //Scanner 보다 빠름 //int N; N = Integer.parseInt(br.readLine()); int [][] map = new int[N][N]; ArrayList houses = new ArrayList(); for (int i = 0; i < N; i++){ String str = br.readLine(); for (int j = 0; j < N; j++){ map[i][j] = Integer.parseInt(String.valueOf(str.charAt(j))); } } br.close(); houses = dfs(map, houses); int num = houses.size(); System.out.println(num); Collections.sort(houses); houses.forEach(house -> System.out.println(house)); } static ArrayList dfs(int[][] map, ArrayList houses) throws Exception { Stack stack = new Stack<>(); int[][] visited = new int[N][N]; int[] temp; int[][] delta = {{-1, 0},{1, 0}, {0,-1},{0, 1}}; while(true) { temp = find_start(map, visited); if (temp[0] == -1 && temp[1] == -1) { break; } stack.push(temp); int ti = temp[0]; int tj = temp[1]; visited[ti][tj] = 1; int num = 1; while (!stack.isEmpty()){ temp = stack.pop(); ti = temp[0]; tj = temp[1]; for (int[] d : delta) { int ni = ti + d[0]; int nj = tj + d[1]; if (0 <= ni && ni < N && 0 <= nj && nj < N && visited[ni][nj] == 0 && map[ni][nj]==1) { stack.push(new int[] {ni, nj}); num++; visited[ni][nj] = 1; ti = ni; tj = nj; } } } houses.add(num); } return houses; } static int[] find_start(int[][] map, int[][] visited) { for(int i = 0; i< N; i++) { for (int j = 0; j

제출 #26944

제출 정보

제출 결과

제출 코드

찾은 반례

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력:

입력:

오답 출력:

정답 출력: