[백준] 연결요소의 개수 (java) Dfs

2020-02-13

연결요소의 개수 (백준 > 11724 > Dfs_Bfs)

문제 링크

문제설명

방향 없는 그래프가 주어졌을 때, 연결 요소 (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


풀이

package Dfs_Bfs;

import java.util.*;

public class beak_연결요소의개수 {

    static void dfs(int arr[][], int idx, boolean check[]){
        //돌고 나오면서 true로 만든다
        check[idx] = true;
        
        for(int k=1; k<arr.length; k++){
            if(arr[idx][k]==1 && check[k] ==false)
                dfs(arr,k,check);
        }
    }
    
    public static void main(String args[]){
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int k = sc.nextInt();
        
        int arr[][] = new int[n+1][n+1];
        
        for(int i=0; i<k; i++){
            int a = sc.nextInt();
            int b = sc.nextInt();
            arr[a][b] = 1;
            arr[b][a] = 1;
        }
        
        int answer = 0;
        
        boolean check[] = new boolean[n+1];
        for(int i=1;i<=n; i++){
                if(check[i] == false){
                    dfs(arr,i,check);
                    answer++;                    
            }
        }
        
        System.out.println(answer);
    }
	
}


후기 (40min)

이거봐!! dfs 안한지 조금 됐다고 그새 check[]쓰는거 까먹고, 숫자는 하나니까 배열은 한번만 돌려도 되는 것을…으아