[백준 4195번] 친구 네트워크

[백준 4195번] 친구 네트워크

www.acmicpc.net/problem/4195

4195번: 친구 네트워크

 

 

 문제

 민혁이는 소셜 네트워크 사이트에서 친구를 만드는 것을 좋아하는 친구이다. 우표를 모으는 취미가 있듯이, 민혁이는 소셜 네트워크 사이트에서 친구를 모으는 것이 취미이다.

 어떤 사이트의 친구 관계가 생긴 순서대로 주어졌을 때, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.

 친구 네트워크란 친구 관계만으로 이동할 수 있는 사이를 말한다.

 

 입력

 첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스의 첫째 줄에는 친구 관계의 수 F가 주어지며, 이 값은 100,000을 넘지 않는다. 다음 F개의 줄에는 친구 관계가 생긴 순서대로 주어진다. 친구 관계는 두 사용자의 아이디로 이루어져 있으며, 알파벳 대문자 또는 소문자로만 이루어진 길이 20 이하의 문자열이다.

 

 출력

 친구 관계가 생길 때마다, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.

 

 

 풀이

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

unordered_map<string, string> w;
unordered_map<string, int> cnt;
string u_find(string k){
	if(w[k] == k) return k;
	else return w[k] = u_find(w[k]);
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
	
	int T; cin >> T;
	for(int q=0;q<T;q++){
		w.clear(); cnt.clear();
		
		int n; cin >> n;
		for(int i=0;i<n;i++){
			string s1, s2; cin >> s1 >> s2;
			if(w.find(s1) == w.end()){
				w[s1] = s1;
				cnt[s1] = 1;
			}
			else s1 = u_find(s1);
			
			if(w.find(s2) == w.end()){
				w[s2] = s2;
				cnt[s2] = 1;
			}
			else s2 = u_find(s2);
			
			if(s1 == s2) cout << cnt[s1] << '\n';
			else if(s1 > s2){
				w[s2] = s1;
				cnt[s1] += cnt[s2];
				cout << cnt[s1] << '\n';
			}
			else{
				w[s1] = s2;
				cnt[s2] += cnt[s1];
				cout << cnt[s2] << '\n';
			}
		}
	}
	return 0;
}

 int, int를 대상으로 하던 유니온파인드를 string, string으로 하는 문제이다. 이를 위해 map을 사용했는데 시간을 좀 더 줄이기 위해 정렬을 하지 않는 unordered_map을 사용했다. 먼저 이름이 map구조에 존재하는지 확인하고 없으면 새로 추가해주고, 존재한다면 해당 이름의 머리, 즉 속해있는 친구 네트워크의 우두머리?를 찾는다. 한줄씩 입력받을 때마다 두명의 이름에 대해 수행하게 되고, 이들의 u_find 반환 값이 같다면 같은 네트워크 상에 있다는 의미이므로 해당 우두머리의 cnt값을 출력한다. 만약 다를 경우, 사전순이 빠른 이름에게 합쳐준다. 이러면 끝이다.