[백준, Java] 연결 요소의 개수 구하기 - DFS

11724번: 연결 요소의 개수

문제 보기 ↓

문제

방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.

출력

첫째 줄에 연결 요소의 개수를 출력한다.

예제 입력1

6 5 // 노드 개수, 에지 개수
1 2
2 5
5 1
3 4
4 6

예제 출력1

2

---

문제 이해하기

Step 1

그래프를 인접 리스트로 저장하고 방문 배열 초기화

 

1     →	2	5
2    →	1	5
3    →	4
4    →	3	6
5    →	2	1
6    →	4

↑ 인접리스트  ↓ 방문 배열

1	2	3	4	5	6
F	F	F	F	F	F

 

Step 2

• 1을 시작점으로 설정
• 인접 리스트를 이용해 인접 노드 탐색
• 탐색 후 방문 여부 체크

1	2	3	4	5	6
T	T	F	F	T	F

 

Step 3

• 시작점을 갱신해 방문하지 않은 노드가 없을 때까지 반복
• 모든 노드를 방문하게 되면 탐색 종료

 

최종 결과

모든 노드를 탐색하여 탐색 종료 -> DFS 총 2회 수행
따라서 연결 요소의 개수 == 2

 

코드 구현하기

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.StringTokenizer;

public class Main {
    static ArrayList<Integer>[] A;
    static boolean visited[];
    public static void main(String[] args) throws IOException {
        BufferedReader br=new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st=new StringTokenizer(br.readLine());
        int n=Integer.parseInt(st.nextToken());
        int m=Integer.parseInt(st.nextToken());
        A=new ArrayList[n+1];
        visited=new boolean[n+1];
        for(int i=1;i<n+1;i++){
            A[i]=new ArrayList<>();
        }
        for(int i=0;i<m;i++){
            st=new StringTokenizer(br.readLine());
            int s=Integer.parseInt(st.nextToken());
            int e=Integer.parseInt(st.nextToken());
            A[s].add(e);
            A[e].add(s);
        }
        int count=0;
        for(int i=1;i<n+1;i++){
            if(!visited[i]){
                count++;
                DFS(i);
            }
        }
        System.out.println(count);
    }
    static void DFS(int v){
        if(visited[v])
            return;
        visited[v]=true;
        for(int i:A[v]){
            if(!visited[i])
                DFS(i);
        }
    }
}

---

참고

Do it! 알고리즘 코딩 테스트 -자바 편