[코드트리/INTERMEDIATE LOW] 마을 구분하기 (JAVA)

🔺 문제

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

🔺 코드

import java.util.*;
import java.io.*;
 
public class Main {
    static int[] dx = {1, -1, 0, 0};    // 4방향 (상하좌우)
    static int[] dy = {0, 0, 1, -1};
 
    static int N, cnt;
    static int[][] map;
    static boolean[][] visited;
 
    static PriorityQueue<Integer> pq = new PriorityQueue<>();    // 마을에 있는 사람 수 저장할 우선순위 큐 (자동 오름차순 정렬)
    static StringBuilder sb = new StringBuilder();
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
 
        N = Integer.parseInt(br.readLine());
        map = new int[N][N];
        visited = new boolean[N][N];
 
        for(int i = 0 ; i < N ; i++) {
            st = new StringTokenizer(br.readLine(), " ");
            for(int j = 0 ; j < N ; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }
 
        for(int i = 0 ; i < N ; i++) {
            for(int j = 0 ; j < N ; j++) {
                if(!visited[i][j] && map[i][j] == 1) {    // 방문할 수 있는 경우 (방문한 적 없고, 사람이 있을 때)
                    cnt = 0;
                    dfs(i, j);
                    pq.add(cnt); // 마을 내 사람 수 저장
                }
            }
        }
 
        // 출력하기
        sb.append(pq.size()).append("\n");    // 총 마을의 개수
        while(!pq.isEmpty()) {
            sb.append(pq.poll()).append("\n");
        }
        System.out.println(sb.toString());
    }
 
    private static void dfs(int x, int y) {
        visited[x][y] = true;
        cnt++;
 
        for(int d = 0 ; d < 4 ; d++) { // 4방향 각각에 대해 DFS 탐색
            int nx = x + dx[d];
            int ny = y + dy[d];
 
            // 해당 방향으로 이동 가능할 때, 이동하며 탐색
            if(inRange(nx, ny) && !visited[nx][ny] && map[nx][ny] == 1) {
                dfs(nx, ny);
            }
        }
    }
 
    private static boolean inRange(int x, int y) {
        return (0 <= x && x < N && 0 <= y && y < N);
    }
}

 

 

 

🧩  해결 아이디어

• DFS

- 시간 복잡도 : O(N^2)

 

 

 

 

---

💬 느낀 점

오랜만에 푸니 재밌는 DFS....

 

1회독	2회독	3회독	4회독	5회독
V

---

(참고)

✔ 백준 2583 영역 구하기

[백준 알고리즘] 백준 2583번 영역 구하기 자바(Java)