[Baekjoon/C++] 11724번 - 연결 요소의 개수

[백준] Baekjoon Online Judge

문제로 이동

 

문제

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

 

입력

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

 

출력

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

 

---

풀이

#include <iostream>
#include <vector>
using namespace std;

int N, M, cnt = 0;
bool arr[1001][1001];
bool check[1001];

void dfs(int now);

void dfs(int now) {
    check[now] = true;

    for (int i = 1; i <= N; i++)
        if (!check[i] && arr[now][i])  dfs(i);
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);

    cin >> N >> M;
    for (int i = 0; i < M; i++) {
        int a, b;
        cin >> a >> b;

        arr[a][b] = arr[b][a] = true;
    }

    for (int i = 1; i <= N; i++) {
        if (!check[i]) {
            cnt++;
            dfs(i);
        }
    }

    cout << cnt << '\n';

    return 0;
}

 

    for (int i = 1; i <= N; i++) {
        if (!check[i]) {
            cnt++;
            dfs(i);
        }
    }

연결 요소가 여러 개 있는 경우 한 번의 dfs()함수 호출로 모든 정점을 순회하지 못할 수 있기에 위와 같은 코드를 넣었다.

 

void dfs(int now) {
    check[now] = true;

    for (int i = 1; i <= N; i++)
        if (!check[i] && arr[now][i])  dfs(i);
}

연결된 정점을 모두 방문하는 함수이다. 중복으로 방문하는 것을 피하기 위해 check[]를 사용한다.