[🥇2 / 백준 4195 / 파이썬] 친구 네트워크

 

4195번: 친구 네트워크

 

문제

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

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

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

입력

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

출력

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

 

예제 입력  2 3 Fred Barney Barney Betty Betty Wilma 3 Fred Barney Betty Wilma Barney Betty

예제 출력  2 3 4 2 2 4

 

풀이

 

정점이 숫자가 아닌 문자열로 입력된다.

따라서 기존 유니온 파인드와 구조는 거의 같지만, 딕셔너리를 이용하여 문제를 해결해야 한다.

 

또한 각 뿌리 노드로부터 구성된 그래프의 크기를 알아야 하기 때문에 network 라는 딕셔너리를 하나 더 만들어줬다.

여기서 union을 할 때, network도 합쳐서 뿌리 노드에 저장하면 된다.

 

주의할 점이 있는데, 뿌리 노드가 같은 입력이 들어올 경우에도 크기를 출력해야 한다.

 

import sys
input = sys.stdin.readline

def union(a,b):
    a = find(a)
    b = find(b)
    if a==b:
        return network[a]
    parents[b] = a
    network[a] = network[a] + network[b]
    return network[a]
    
def find(k):
    if k != parents[k]:
        parents[k] = find(parents[k])
    return parents[k]

for _ in range(int(input())):
    parents = dict()
    network = dict()
    for _ in range(int(input())):
        a,b = input().rstrip().split()
        if parents.get(a) == None:
            parents[a] = a
            network[a] = 1
        if parents.get(b) == None:
            parents[b] = b
            network[b] = 1
            
        print(union(a,b))