백준 2667번 : 단지번호붙이기 (Java)

🔗 문제 링크

2667번: 단지번호붙이기

---

📖 풀이 과정

- 재귀를 이용하여 DFS 방식으로 풀었습니다. (BFS방식으로 풀어도 됩니다.)

- 기본적으로 배열의 값이 1이면서, 방문하지 않은 곳을 기준으로 DFS를 사용합니다.

- 이후, DFS 재귀내에서 count 변수를 통해 연결되어 있는 Node를 탐색하여 count 변수를 반환합니다.
  - Main 메소드에서 dfs() 메소드의 시작점 이 곧 count 변수에 포함되기 때문에 초기화를 1로 합니다.
  
- 어차피, 총 단지수는 List의 size()와 같기 때문에 따로 체크하지 않습니다.

- 마지막으로 List에 담은 값들을 정렬하여 출력합니다.

---

💻 코드

import java.io.*;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Main {
    static int n,count;
    static int[] dx = {-1,0,1,0};
    static int[] dy = {0,-1,0,1};
    static int[][] arr;
    static boolean[][] visited;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuffer sb = new StringBuffer();
        List<Integer> answerList = new ArrayList<>();

        n = Integer.parseInt(br.readLine());
        arr = new int[n][n];
        visited = new boolean[n][n];

        for(int i = 0; i< n; i++){
            String input = br.readLine();
            for(int j = 0; j< n; j++){
                arr[i][j] = Integer.parseInt(String.valueOf(input.charAt(j)));
            }
        }

        for(int i = 0; i< n; i++){
            for(int j = 0; j< n; j++){
                if(arr[i][j] == 1 && !visited[i][j]){
                    count = 1;
                    answerList.add(dfs(i,j));
                }
            }
        }
        Collections.sort(answerList);

        sb.append(answerList.size()).append("\n");
        for(Integer i : answerList){
            sb.append(i).append("\n");
        }

        bw.write(sb.toString());
        bw.flush();
        bw.close();
        br.close();
    }

    private static int dfs(int x, int y){
        visited[x][y] = true;

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

            if(nextX >= 0 && nextY >= 0 && nextX < n && nextY < n){
                if(!visited[nextX][nextY] && arr[nextX][nextY] == 1){
                    dfs(nextX,nextY);
                    count += 1;
                }
            }
        }
        return count;
    }
}