[DFS] 백준 - 2667번: 단지번호붙이기 Java 풀이

[공부용이기 때문에 코드가 깔끔하지 않을 수 있습니다!]

 

2667번: 단지번호붙이기

 

문제의 해석은 다음과 같다.

map[n][n]의 지도가 있고 그 위에 집은 1로 표시되고 아무것도 없으면 0으로 표시된다.

그리고 이 집들이 동서남북으로 이루어지면 단지를 이루게 된다.

그래서 총 단지는 몇 개이고 단지마다 집이 몇 개인지 오름차순으로 정렬해서 표시해주면 된다.

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class Main {
    static int n;
    static boolean[][] map;
    static boolean[][] visited;
    static int dx[] = {-1, 1, 0, 0};
    static int dy[] = {0, 0, -1, 1};
    static List<Integer> nums;
    static int c;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        n = Integer.parseInt(st.nextToken());

        map = new boolean[n][n];
        visited = new boolean[n][n];
        nums = new ArrayList<>();
        c = 0;

        for (int i = 0; i < n; i++) {
            st = new StringTokenizer(br.readLine());
            String a = st.nextToken();
            for (int j = 0; j < n; j++) {
                char c = a.charAt(j);
                if (c != '0') {
                    map[i][j] = true;
                }
            }
        }

        int count = 0;

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                if (!visited[i][j] && map[i][j]) {
                    dfs(i, j);
                    count++;
                    nums.add(c);
                    c = 0;
                }
            }
        }

        System.out.println(count);
        Collections.sort(nums);
        for (Integer num : nums) {
            System.out.println(num + 1);
        }
    }

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

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

            if (nowX >= 0 && nowY >= 0 && nowX < n && nowY < n) {
                if (!visited[nowX][nowY] && map[nowX][nowY]) {
                    dfs(nowX, nowY);
                    c++;
                }
            }
        }
    }
}

 

구조를 보면 크게 어렵지 않다.

dfs를 통해 동서남북으로 맵 내에 방문하지 않았고 다음 좌표에 집에 있을 경우 dfs 반복을 시켜주면서 카운트를 올려주면 된다.

이 카운트는 단지 내 집의 개수를 표현해줄 것이고 메인코드에서 dfs를 마치면 단지 개수를 ++해주고 nums에 집 개수를 넣어준 후 집을 세는 c는 0으로 초기화를 해준다.

 

그리고 마지막으로 단지 개수와 집 개수를 프린트를 해주는데 이때 DFS 시 첫 시작하는 좌표의 집 개수는 세지 않았으니 +1을 해서 출력하면 된다.