4-dhlee777

[dhlee777]

🔗 문제 링크

https://www.acmicpc.net/problem/2606

✔️ 소요된 시간

2시간

✨ 수도 코드

이문제는 단순한 bfs를 이용해 풀 수 있지만 유니온파인드를 이용해서도 한번 풀어보았다.

첫번째풀이

첫번재풀이는 그냥 단순히 bfs를 이용하여 풀었다. 1번 컴퓨터와 연결된 컴퓨터는 모조리 감염되는것이므로 bfs를 이용하여 1번컴퓨터와 연결된 모든 컴퓨터를 다 탐색해주며 카운트 하면 된다. 여기서 주의해야할점은 맨처음에 시작정점을 방문했다고 표시해주는것! (visited[a]=1) 맨처음 이부분을 놓쳤어서 계속 이상한 답이 나왔다..

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
vector<int>com_edge[101];    //컴퓨터간의 간선의 정보를 저장하는 이차원벡터
int visited[101];   //감염된 컴퓨터인지 확인하기위한 배열
queue<int> q;
int coun = 0;
void bfs(int a) {     //bfs 를 위한 함수
	q.push(a);
	visited[a] = 1;
	while (!q.empty()) {
		int com = q.front();
		q.pop();
		for (int i = 0; i < com_edge[com].size(); i++) {
			int com_linked = com_edge[com][i];
			if (!visited[com_linked])
			{
				q.push(com_linked);
				coun++;            //방문되지않은 연결된 컴퓨터일경우 카운트
				visited[com_linked] = 1;
			}
		}
	}

}
int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int com_num, edge, a, b;
	cin >> com_num >> edge;
	for (int i = 0; i < edge; i++) {
		cin >> a >> b;
		com_edge[a].push_back(b);
		com_edge[b].push_back(a);
	}
	bfs(1);
	cout << coun;
	return 0;
}

두번째풀이

두번째풀이는 유니온파인드를 사용하여 풀어주었다. 1번컴퓨터와 연결된 컴퓨터는 다 감염되는것이므로 유니온 파인드를 이용해 루트노드를 1번 컴퓨터로 가지는 컴퓨터는(1번 컴퓨터와 같은 집합의 컴퓨터)모두 감염된 컴퓨터라고 생각했다.
전체적인 흐름은

1. add함수를 통해 컴퓨터 집합을 만들어준다.
2. for문을 통해 모든 컴퓨터의 루트 컴퓨터를 확인해주며 만약 루트컴퓨터가 1일경우 감염된 컴퓨터와 같은 집합이므로 카운트해준다.
3. 1번컴퓨터를 통해 감염된 컴퓨터의 수를 카운트 하는것이므로 1번컴퓨터는 카운트에서 제거해준다.

#include<iostream>
using namespace std;
int parent [101];      //어떤컴퓨터의 부모컴퓨터를 저장하는 배열
int find_parent(int a) {      //루트 컴퓨터를 찾는 함수
	if (a == parent[a]) return a;
	else return parent[a] = find_parent(parent[a]);

}
void add(int a, int b) {    //두개의 컴퓨터가 속한 각각의 집합을 합치는 함수
	a = find_parent(a);
	b = find_parent(b);
	if (a > b) parent[a] = b;
	else parent[b] = a;
}
int main(void) {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL);
	int com_num, edge,a,b;       
	cin >> com_num >> edge;
	for (int i = 1; i <= com_num; i++) {   //제일처음 자신의 부모는 자신이기에 초기화해준다.
		parent[i] = i;
	}
	for (int i = 0; i < edge; i++) {
		cin >> a >> b;
		add(a, b);
	}
	int coun = 0;
	for (int i = 1; i <= com_num; i++) {     // 어떤컴퓨터의 루트컴퓨터가 1이라면 1과 연결된 같은 집합이므로 숫자를 카운트한다.
		if (find_parent(i) == 1) coun++;
	}
	
	cout << coun-1;    //1번 컴퓨터에의해 감염된 컴퓨터의 수를 세는것이므로 1번컴퓨터를 제외해준다.
	return 0;
}

📚 새롭게 알게된 내용

단순한 bfs문제이지만 유니온파인드를 이용해서 풀어보니 뭔가 시간복잡도가 더 크고 비효율적인것 같긴하지만 색달랐다.

[InSange]

유니온 파인드로 접근했을 때는 입력으로 주어지는 a, b값을 일일이 트리 높이를 올려치기하면서 비교해야되다보니 깊이의 크기만큼 시간 복잡도가 주어지는 것 같습니다!
그에 반해 bfs는 연결되어 있는 노드들이 연계되는 방식이다보니 한 번 비교해주면 쭉 연결되기때문에 차이가 어느정도 나는 것 같네요!
추후에는 시간 복잡도까지 명시해주시면 보기 더 좋을 것 같습니당!!
두 가지 접근법으로 푸는 방법 너무 좋았습니다~

[dhlee777]

추후에는 한번 시간복잡도를 명시해보겠습니다!

[seongwon030]

바이러스 문제에 유니온파인드를 적용해서 풀 수 있었네요. 같은 집합인지 확인하는 과정에서 루트 노드가 1일때 감염된 것이므로 같은 집합으로 분류하여 카운트하는 방식 너무 잘 봤습니다 ㅎㅎ

[yuyu0830]

2번째 접근은 생각 못했네요. 재밌게 잘 봤습니다 :)
나머지는 전부 잘 이해했는데 혹시 count 변수를 coun 이라고 쓰는 이유가 있나요? cnt나 count, c 까지는 봤는데 coun 은 또 처음이네요 ㅎㅎ

[dhlee777]

count 라 선언해주니까 계속 키워드가 모호하다고 해서 t를 빼서 coun으로 바꿔줬는데 듣고보니 cnt가 더 직관적인거 같습니다 다음부터는 cnt로 선언해줘야겠네요 !