[Silver II] 연결 요소의 개수 - 11724

문제 링크

문제 링크

성능 요약

메모리: 6524 KB, 시간: 76 ms

문제 설명

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

using namespace std;

void bfs(int st, const vector<vector<int>>& node, vector<bool>& visited) {
	queue<int> q;
	q.push(st);
	visited[st] = true;

	while (!q.empty()) {
		int cur = q.front();
		q.pop();

		for (int i = 0; i < node[cur].size(); i++) {
			int next = node[cur][i];
			if (visited[next]) continue;
			visited[next] = true;
			q.push(next);
		}
	}
}

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

	int N, M;
	cin >> N >> M;

	vector<vector<int>> node(N + 1);
	vector<bool> visited(N + 1, false);

	int res = 0;

	for (int i = 0; i < M; i++) {
		int from, to;
		cin >> from >> to;
		node[from].push_back(to);
		node[to].push_back(from);
	}

	for (int i = 1; i <= N; i++) {
		if (!visited[i]) {
			bfs(i, node, visited);
			res++;
		}
	}

	cout << res << "\n";

	return 0;
}