[DFS] 백준 11724번 : 연결 요소의 개수 - Java

https://www.acmicpc.net/problem/2606

이번 문제는 위의 문제와 거의 같다. (풀이 참고 : https://kimjunhyun.tistory.com/134)

 

✅ 문제

https://www.acmicpc.net/problem/11724

 

접근방법

1. 연결된 요소의 개수 -> DFS / BFS

2. 서로 연결되었다는 정보를 어떻게 하나의 자료구조로 통합할까?

3. 이미 방문한 지점을 다시 방문하지 않으려면 어떤 자료구조를 사용해야할까?

4. 어디에서 DFS를 시작할 것인가?

 

🔑 풀이

import java.io.*;
import java.util.*;

// DFS
public class Main {

    final static int MAX = 1000 + 10; // N이 가질 수 있는 최댓값
    static boolean[][] graph;
    static boolean[] visited; // 재방문 방지 배열
    static int N, M;
    static int answer;
    static StringBuilder sb = new StringBuilder();

    static void dfs(int idx) {
        visited[idx] = true;
        for (int i = 1; i <= N ; i++) {
            if(!visited[i] && graph[idx][i])
                dfs(i);
        }
    }

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        StringTokenizer st = new StringTokenizer(br.readLine());
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());

        // 1. graph에 연결 정보 채우기
        graph = new boolean[MAX][MAX];
        visited = new boolean[MAX];

        for (int i = 0; i < M; i++) {
            st = new StringTokenizer(br.readLine());
            int u = Integer.parseInt(st.nextToken());
            int v = Integer.parseInt(st.nextToken());
            graph[u][v] = true;
            graph[v][u] = true;
        }

        // 2. dfs(재귀함수 호출)
        for (int i = 1; i <= N; i++) {
            if(!visited[i]) {
                dfs(i);
                answer++;
            }
        }

        // 3. 출력
        bw.write(String.valueOf(answer));

        bw.close();
        br.close();

    }

}