문제
민혁이는 소셜 네트워크 사이트에서 친구를 만드는 것을 좋아하는 친구이다. 우표를 모으는 취미가 있듯이, 민혁이는 소셜 네트워크 사이트에서 친구를 모으는 것이 취미이다.
어떤 사이트의 친구 관계가 생긴 순서대로 주어졌을 때, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.
친구 네트워크란 친구 관계만으로 이동할 수 있는 사이를 말한다.
입력
첫째 줄에 테스트 케이스의 개수가 주어진다. 각 테스트 케이스의 첫째 줄에는 친구 관계의 수 F가 주어지며, 이 값은 100,000을 넘지 않는다. 다음 F개의 줄에는 친구 관계가 생긴 순서대로 주어진다. 친구 관계는 두 사용자의 아이디로 이루어져 있으며, 알파벳 대문자 또는 소문자로만 이루어진 길이 20 이하의 문자열이다.
출력
친구 관계가 생길 때마다, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
def find(x):
if parents[x] == x:
return x
else:
parents[x] = find(parents[x])
return parents[x]
def union(a,b):
a = find(a)
b = find(b)
if a!=b:
parents[b]=a
number[a]+=number[b]
n = int(input())
for _ in range(n):
f = int(input())
parents = {}
number = {}
for _ in range(f):
a,b=map(str,input().split())
if a not in parents:
parents[a]=a
number[a]=1
if b not in parents:
parents[b]=b
number[b]=1
union(a,b)
print(number[find(a)])
느낀 점
스스로 다 풀지 못하고, 도움을 빌렸기 때문에 제출하지 않았다. 나중에 다시 풀어볼 예정.
유니온파인드 문제가 아직 안 익숙해서 그런가, 이 개념이 아직 희미한 상태에서 문제를 보면 문제 아이디어가 안 떠오른다기보다, 문제 워딩 자체가 모호하고 어색하게 느껴진다.
“친구 네트워크란 친구 관계만으로 이동할 수 있는 사이를 말한다.”
“친구 관계가 생길 때마다, 두 사람의 친구 네트워크에 몇 명이 있는지 구하는 프로그램을 작성하시오.”
등…
문과 출신이라서 그런가 워딩 자체에 집중하는 경향이 있다.
이런 습관을 배재하고 최대한 어떤 알고리즘을 요구하는건지만 머릿속에 떠올리는 연습을 해야겠다.
문제 Key Point
유니온파인드는 find로 부모로 재귀하기 때문에 기록해야 되는 수를 부모쪽에 넣어두고, 연결되어 있는 나머지는 부모를 가리키는것이 포인트인 것 같다.