https://www.acmicpc.net/problem/4195
\n문제
민혁이는 소셜 네트워크 사이트에서 친구를 만드는 것을 좋아하는 친구이다. 우표를 모으는 취미가 있듯이, 민혁이는 소셜 네트워크 사이트에서 친구를 모으는 것이 취미이다.
\n어떤 사이트의 친구 관계가 생긴 순서대로 주어졌을 때, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.
\n친구 네트워크란 친구 관계만으로 이동할 수 있는 사이를 말한다.
\n입력
첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스의 첫째 줄에는 친구 관계의 수 F가 주어지며, 이 값은 100,000을 넘지 않는다. 다음 F개의 줄에는 친구 관계가 생긴 순서대로 주어진다. 친구 관계는 두 사용자의 아이디로 이루어져 있으며, 알파벳 대문자 또는 소문자로만 이루어진 길이 20 이하의 문자열이다.
\n출력
친구 관계가 생길 때마다, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.
\n유니온 파인드를 조금 응용하면 풀 수 있는 문제였다. 기존의 유니온 파인드 문제와 달리 이번에는 입력으로 문자열이 들어왔는데, 이 문자열을 맵에 고유한 인덱스 번호를 생성해서 함께 맵핑해두고 실제 유니온, 파인드 연산은 key에 해당하는 value 값으로 연산했다.
\n' '\n#include <cstdio>\n#include <string>\n#include <map>\n\nusing namespace std;\n\nint parent[200001];\nint cnt[200001];\n\nint findRoot(int x) {\n if (parent[x] == x) return x;\n else return parent[x] = findRoot(parent[x]);\n}\n\nint merge(int a, int b) {\n a = findRoot(a);\n b = findRoot(b);\n\n if (a == b) return cnt[a];\n\n if (a < b) {\n parent[b] = a;\n cnt[a] += cnt[b];\n }\n else {\n parent[a] = b;\n cnt[b] += cnt[a];\n }\n\n return a < b ? cnt[a] : cnt[b];\n}\n\nint main() {\n int T;\n scanf(\"%d\", &T);\n\n while (T--) {\n map<string, int> network;\n int friendIdx = 1;\n int F;\n scanf(\"%d\", &F);\n\n for (int i = 1; i <= 200000; i++) {\n parent[i] = i;\n cnt[i] = 1;\n }\n\n for (int i = 0; i < F; i++) {\n char firstName[21];\n char secondName[21];\n int idxFirstName;\n int idxSecondName;\n\n scanf(\"%s %s\", firstName, secondName);\n\n auto iterFirstName = network.find(firstName);\n auto iterSecondName = network.find(secondName);\n\n if (iterFirstName == network.end()) {\n network.insert({ firstName, friendIdx });\n idxFirstName = friendIdx++;\n }\n else idxFirstName = (*iterFirstName).second;\n\n if (iterSecondName == network.end()) {\n network.insert({ secondName, friendIdx });\n idxSecondName = friendIdx++;\n }\n else idxSecondName = (*iterSecondName).second;\n\n int ans = merge(idxFirstName, idxSecondName);\n\n printf(\"%d\\n\", ans);\n }\n }\n}