BOJ 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

예제 입력 2

6 8
1 2
2 5
5 1
3 4
4 6
5 4
2 4
2 3

예제 출력 2

1

DFS 로 풀기

#include <iostream>
#include <vector>
#define IO std::cin.tie(NULL), std::ios_base::sync_with_stdio(false)

#define MAX 1001

int N, M;
std::vector<int> V[MAX];
bool visited[MAX];

void DFS(int s) {
  visited[s] = true;

  for (auto &next : V[s]) {
    if (!visited[next]) {
      DFS(next);
    }
  }
}

void solve() {
  std::cin >> N >> M;

  int u, v;
  for (int i = 0; i < M; i++) {
    std::cin >> u >> v;
    V[u].push_back(v);
    V[v].push_back(u);
  }

  int count = 0;
  for (int i = 1; i <= N; i++) {
    if (!visited[i]) {
      count++;
      DFS(i);
    }
  }

  std::cout << count;
}

int main() {
  IO;
  solve();
  return 0;
}

BFS로 풀기

#include <iostream>
#include <queue>
#include <vector>
#define IO std::cin.tie(NULL), std::ios_base::sync_with_stdio(false)

#define MAX 1001

int N, M;
std::vector<int> V[MAX];
bool visited[MAX];

void BFS(int s) {
  std::queue<int> Q;
  visited[s] = true;

  Q.push(s);

  while (!Q.empty()) {
    int qs = Q.front();
    Q.pop();

    for (auto &next : V[qs]) {
      if (!visited[next]) {
        visited[next] = true;
        Q.push(next);
      }
    }
  }
}

void solve() {
  std::cin >> N >> M;

  int u, v;
  for (int i = 0; i < M; i++) {
    std::cin >> u >> v;
    V[u].push_back(v);
    V[v].push_back(u);
  }

  int count = 0;
  for (int i = 1; i <= N; i++) {
    if (!visited[i]) {
      count++;
      BFS(i);
    }
  }

  std::cout << count;
}

int main() {
  IO;
  solve();
  return 0;
}