백준 11724번 연결 요소의 개수 [JAVA]

문제

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

 

 

입력

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

 

 

출력

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

 

 

예제

 

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
 
public class Main_BJ_11724_연결요소의개수 {
    static int N,M, result;
    static boolean []visited;
    static int [][]input;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st ;
        
        st = new StringTokenizer(br.readLine()," ");
        
        N = Integer.parseInt(st.nextToken());
        M = Integer.parseInt(st.nextToken());
        
        input = new int [N+1][N+1];
        visited = new boolean[N+1];
        
        for(int i=0;i<M;i++) {
            st = new StringTokenizer(br.readLine()," ");
            int a = Integer.parseInt(st.nextToken());
            int b = Integer.parseInt(st.nextToken());
            
            input[a][b] = 1; 
            input[b][a]= 1;
        }
        
        for(int i=1;i<N+1;i++) {
            if(!visited[i]) {
                bfs(i); result++;
            }
        }
        System.out.println(result);
    }
    
    public static void bfs(int i) {
        visited[i] = true;
        
        for(int j=1;j<=N;j++) {
            if(!visited[j] && input[i][j]==1) {
                bfs(j);
            }
        }
    }
 
}

 

구현 방법

입력받은 간선의 양 끝점인 a 와 b를 input 배열에 넣어주었습니다. 그 후, for 문을 돌려 방문 체크를 하면서 만약 방문하지 않는 곳이라면 연결 요소의 개수를 +1 해주고 bfs() 함수에서 해당 간선과 연결된 곳을 for문 통해 확인해 주었습니다.