728x90
문제
방향 없는 그래프가 주어졌을 때, 연결 요소 (Connected Component)의 개수를 구하는 프로그램을 작성하시오.
입력
첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 0 ≤ M ≤ N×(N-1)/2) 둘째 줄부터 M개의 줄에 간선의 양 끝점 u와 v가 주어진다. (1 ≤ u, v ≤ N, u ≠ v) 같은 간선은 한 번만 주어진다.
출력
첫째 줄에 연결 요소의 개수를 출력한다.
풀이과정
1. BFS와 DFS를 이용하여 문제를 해결했습니다.
2. 연결 요소의 개념을 알면 그래프 탐색을 통해 쉽게 풀 수 있는 문제였습니다.
여기서 연결 요소는 그래프에서 연결된 덩어리를 뜻합니다.
예제 입력 1번 처럼 그래프를 연결하면 연결 요소가 2개가 나옵니다.
코드
BFS 풀이
#include<iostream>
#include<queue>
#include<vector>
#define endl "\n"
using namespace std;
vector<int> graph[1001];
bool visit[1001];
int N, M;
void BFS(int index)
{
queue<int> q;
q.push(index);
visit[index] = true;
while (!q.empty())
{
int index = q.front();
q.pop();
for (int i = 0; i < graph[index].size(); i++)
{
if (visit[graph[index][i]]) continue;
visit[graph[index][i]] = true;
q.push(graph[index][i]);
}
}
}
void Answer()
{
cin >> N >> M;
int u, v;
int answer = 0;
for (int i = 0; i < M; i++)
{
cin >> u >> v;
graph[u].push_back(v);
graph[v].push_back(u);
}
for (int i = 1; i <= N; i++)
{
if (visit[i]) continue;
BFS(i);
answer++;
}
cout << answer;
}
int main()
{
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
Answer();
return 0;
}
DFS 풀이
#include<iostream>
#include<vector>
#define endl "\n"
using namespace std;
vector<int> graph[1001];
bool visit[1001];
int N, M;
void DFS(int index)
{
visit[index] = true;
for (int i = 0; i < graph[index].size(); i++)
{
if (visit[graph[index][i]]) continue;
DFS(graph[index][i]);
}
}
void Answer()
{
cin >> N >> M;
int u, v;
int answer = 0;
for (int i = 0; i < M; i++)
{
cin >> u >> v;
graph[u].push_back(v);
graph[v].push_back(u);
}
for (int i = 1; i <= N; i++)
{
if (visit[i]) continue;
DFS(i);
answer++;
}
cout << answer;
}
int main()
{
ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
Answer();
return 0;
}
728x90