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

문제

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

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

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

입력

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

출력

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

---

입력이 이름으로, 즉 string으로 들어온다는 점만 주의하면 되는 disjoint-set 문제이다.

string으로 이름이 들어온다는 점에서 바로 map을사용하겠구나를 알아채고 map 사용법을 본 후 구현했다.

map에 이름이 있는지 알아내기 위해 map.find 함수를 쓰고, 없으면 map.end()와 같아짐을 이용해 새 값을 insert해준다. 

그 후론 이름에 해당하는 인덱스가 생기는 꼴이므로 union과 find를 적절히 사용해준다.

 

시간초과가 떴었는데 최적화가 덜 됐나 생각했는데, cin과 cout의 속도가 너무 느린 탓이었다.

ios::sync_with_stdio(false);
cin.tie(0);

를 꼭 써주도록 하자

 

 

코드 원본 : https://github.com/chosh95/STUDY/blob/master/BaekJoon/2020/%EC%B9%9C%EA%B5%AC%20%EB%84%A4%ED%8A%B8%EC%9B%8C%ED%81%AC(4195).cpp

chosh95/STUDY

C++ 코드

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <map>
using namespace std;
vector<int> parent, setRank, setSize;
int C, F;

int Find(int u) {
	if (parent[u] == u) return u;
	return parent[u] = Find(parent[u]);
}

void Union(int u, int v) {
	u = Find(u), v = Find(v);
	if (u == v) return;
	if (setRank[u] > setRank[v]) swap(u, v);
	parent[u] = v;
	if (setRank[u] == setRank[v]) setRank[v]++;
	setSize[v] += setSize[u];
}

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

	cin >> C;
	while (C--) {
		cin >> F;

		parent.resize(2*F + 1);
		for (int i = 0; i < 2*F + 1; i++) parent[i] = i;
		setSize.assign(2*F + 1, 1);
		setRank.assign(2*F + 1, 1);

		map<string, int> m;

		int cnt = 0;
		string a, b;
		for (int i = 0; i < F; i++) {
			cin >> a >> b;
			if (m.find(a) == m.end()) {
				m.insert({ a,cnt++ });
			}
			if (m.find(b) == m.end()) {
				m.insert({ b,cnt++ });
			}
			Union(m[a], m[b]);
			cout << setSize[Find(m[a])] << "\n";
		}
	}
}