백준 온라인 저지, 분리집합 / 4195번: 친구네트워크 (파이썬 / 백준 골드문제)

문제

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

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

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

입력

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

출력

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

제한

예제 입력 1

2
3
Fred Barney
Barney Betty
Betty Wilma
3
Fred Barney
Betty Wilma
Barney Betty

예제 출력 1

2
3
4
2
2
4

힌트

W3sicHJvYmxlbV9pZCI6IjQxOTUiLCJwcm9ibGVtX2xhbmciOiIwIiwidGl0bGUiOiJcdWNlNWNcdWFkNmMgXHViMTI0XHVkMmI4XHVjNmNjXHVkMDZjIiwiZGVzY3JpcHRpb24iOiI8cD5cdWJiZmNcdWQ2MDFcdWM3NzRcdWIyOTQgXHVjMThjXHVjMTVjIFx1YjEyNFx1ZDJiOFx1YzZjY1x1ZDA2YyZuYnNwO1x1YzBhY1x1Yzc3NFx1ZDJiOFx1YzVkMFx1YzExYyBcdWNlNWNcdWFkNmNcdWI5N2MgXHViOWNjXHViNGRjXHViMjk0IFx1YWM4M1x1Yzc0NCBcdWM4OGJcdWM1NDRcdWQ1NThcdWIyOTQgXHVjZTVjXHVhZDZjXHVjNzc0XHViMmU0LiBcdWM2YjBcdWQ0NWNcdWI5N2MgXHViYWE4XHVjNzNjXHViMjk0IFx1Y2RlOFx1YmJmOFx1YWMwMCBcdWM3ODhcdWI0ZWZcdWM3NzQsIFx1YmJmY1x1ZDYwMVx1Yzc3NFx1YjI5NCBcdWMxOGNcdWMxNWMgXHViMTI0XHVkMmI4XHVjNmNjXHVkMDZjJm5ic3A7XHVjMGFjXHVjNzc0XHVkMmI4XHVjNWQwXHVjMTFjIFx1Y2U1Y1x1YWQ2Y1x1Yjk3YyBcdWJhYThcdWM3M2NcdWIyOTQgXHVhYzgzXHVjNzc0IFx1Y2RlOFx1YmJmOFx1Yzc3NFx1YjJlNC48XC9wPlxyXG5cclxuPHA+XHVjNWI0XHViNWE0IFx1YzBhY1x1Yzc3NFx1ZDJiOFx1Yzc1OCBcdWNlNWNcdWFkNmMgXHVhZDAwXHVhY2M0XHVhYzAwIFx1YzBkZFx1YWUzNCBcdWMyMWNcdWMxMWNcdWIzMDBcdWI4NWMgXHVjOGZjXHVjNWI0XHVjODRjXHVjNzQ0IFx1YjU0YywgXHViNDUwIFx1YzBhY1x1Yjc4Y1x1Yzc1OCBcdWNlNWNcdWFkNmMgXHViMTI0XHVkMmI4XHVjNmNjXHVkMDZjXHVjNWQwIFx1YmE4NyBcdWJhODVcdWM3NzQgXHVjNzg4XHViMjk0XHVjOWMwIFx1YWQ2Y1x1ZDU1OFx1YjI5NCBcdWQ1MDRcdWI4NWNcdWFkZjhcdWI3YThcdWM3NDQgXHVjNzkxXHVjMTMxXHVkNTU4XHVjMmRjXHVjNjI0LjxcL3A+XHJcblxyXG48cD5cdWNlNWNcdWFkNmMgXHViMTI0XHVkMmI4XHVjNmNjXHVkMDZjXHViNzgwIFx1Y2U1Y1x1YWQ2YyBcdWFkMDBcdWFjYzRcdWI5Y2NcdWM3M2NcdWI4NWMgXHVjNzc0XHViM2Q5XHVkNTYwIFx1YzIxOCBcdWM3ODhcdWIyOTQgXHVjMGFjXHVjNzc0XHViOTdjIFx1YjlkMFx1ZDU1Y1x1YjJlNC48XC9wPlxyXG4iLCJpbnB1dCI6IjxwPlx1Y2NhYlx1YzlmOCBcdWM5MDRcdWM1ZDAgXHVkMTRjXHVjMmE0XHVkMmI4IFx1Y2YwMFx1Yzc3NFx1YzJhNFx1Yzc1OCBcdWFjMWNcdWMyMThcdWFjMDAgXHVjOGZjXHVjNWI0XHVjOWM0XHViMmU0LiBcdWFjMDEgXHVkMTRjXHVjMmE0XHVkMmI4IFx1Y2YwMFx1Yzc3NFx1YzJhNFx1Yzc1OCBcdWNjYWJcdWM5ZjggXHVjOTA0XHVjNWQwXHViMjk0IFx1Y2U1Y1x1YWQ2YyBcdWFkMDBcdWFjYzRcdWM3NTggXHVjMjE4IEZcdWFjMDAgXHVjOGZjXHVjNWI0XHVjOWMwXHViYTcwLCBcdWM3NzQgXHVhYzEyXHVjNzQwIDEwMCwwMDBcdWM3NDQgXHViMTE4XHVjOWMwIFx1YzU0YVx1YjI5NFx1YjJlNC4gXHViMmU0XHVjNzRjIEZcdWFjMWNcdWM3NTggXHVjOTA0XHVjNWQwXHViMjk0IFx1Y2U1Y1x1YWQ2YyBcdWFkMDBcdWFjYzRcdWFjMDAgXHVjMGRkXHVhZTM0IFx1YzIxY1x1YzExY1x1YjMwMFx1Yjg1YyBcdWM4ZmNcdWM1YjRcdWM5YzRcdWIyZTQuIFx1Y2U1Y1x1YWQ2YyBcdWFkMDBcdWFjYzRcdWIyOTQgXHViNDUwIFx1YzBhY1x1YzZhOVx1Yzc5MFx1Yzc1OCBcdWM1NDRcdWM3NzRcdWI1MTRcdWI4NWMgXHVjNzc0XHViOGU4XHVjNWI0XHVjODM4IFx1Yzc4OFx1YzczY1x1YmE3MCwgXHVjNTRjXHVkMzBjXHViY2IzIFx1YjMwMFx1YmIzOFx1Yzc5MCBcdWI2MTBcdWIyOTQgXHVjMThjXHViYjM4XHVjNzkwXHViODVjXHViOWNjIFx1Yzc3NFx1YjhlOFx1YzViNFx1YzljNCBcdWFlMzhcdWM3NzQgMjAgXHVjNzc0XHVkNTU4XHVjNzU4IFx1YmIzOFx1Yzc5MFx1YzVmNFx1Yzc3NFx1YjJlNC48XC9wPlxyXG4iLCJvdXRwdXQiOiI8cD5cdWNlNWNcdWFkNmMgXHVhZDAwXHVhY2M0XHVhYzAwIFx1YzBkZFx1YWUzOCBcdWI1NGNcdWI5YzhcdWIyZTQsIFx1YjQ1MCBcdWMwYWNcdWI3OGNcdWM3NTggXHVjZTVjXHVhZDZjIFx1YjEyNFx1ZDJiOFx1YzZjY1x1ZDA2Y1x1YzVkMCBcdWJhODcgXHViYTg1XHVjNzc0IFx1Yzc4OFx1YjI5NFx1YzljMCBcdWFkNmNcdWQ1NThcdWIyOTQgXHVkNTA0XHViODVjXHVhZGY4XHViN2E4XHVjNzQ0IFx1Yzc5MVx1YzEzMVx1ZDU1OFx1YzJkY1x1YzYyNC48XC9wPlxyXG4iLCJoaW50IjoiIiwib3JpZ2luYWwiOiIwIiwiaHRtbF90aXRsZSI6IjAiLCJwcm9ibGVtX2xhbmdfdGNvZGUiOiJLb3JlYW4ifSx7InByb2JsZW1faWQiOiI0MTk1IiwicHJvYmxlbV9sYW5nIjoiMSIsInRpdGxlIjoiVmlydHVhbCBGcmllbmRzIiwiZGVzY3JpcHRpb24iOiI8cD5UaGVzZSBkYXlzLCB5b3UgY2FuIGRvIGFsbCBzb3J0cyBvZiB0aGluZ3Mgb25saW5lLiBGb3IgZXhhbXBsZSwgeW91IGNhbiB1c2UgdmFyaW91cyB3ZWJzaXRlcyB0byBtYWtlIHZpcnR1YWwgZnJpZW5kcy4gRm9yIHNvbWUgcGVvcGxlLCBncm93aW5nIHRoZWlyIHNvY2lhbCBuZXR3b3JrICh0aGVpciBmcmllbmRzLCB0aGVpciBmcmllbmRzJiMzOTsgZnJpZW5kcywgdGhlaXIgZnJpZW5kcyYjMzk7IGZyaWVuZHMmIzM5OyBmcmllbmRzLCBhbmQgc28gb24pLCBoYXMgYmVjb21lIGFuIGFkZGljdGl2ZSBob2JieS4gSnVzdCBhcyBzb21lIHBlb3BsZSBjb2xsZWN0IHN0YW1wcywgb3RoZXIgcGVvcGxlIGNvbGxlY3QgdmlydHVhbCBmcmllbmRzLjxcL3A+XHJcblxyXG48cD5Zb3VyIHRhc2sgaXMgdG8gb2JzZXJ2ZSB0aGUgaW50ZXJhY3Rpb25zIG9uIHN1Y2ggYSB3ZWJzaXRlIGFuZCBrZWVwIHRyYWNrIG9mIHRoZSBzaXplIG9mIGVhY2ggcGVyc29uJiMzOTtzIG5ldHdvcmsuPFwvcD5cclxuXHJcbjxwPkFzc3VtZSB0aGF0IGV2ZXJ5IGZyaWVuZHNoaXAgaXMgbXV0dWFsLiBJZiBGcmVkIGlzIEJhcm5leSYjMzk7cyBmcmllbmQsIHRoZW4gQmFybmV5IGlzIGFsc28gRnJlZCYjMzk7cyBmcmllbmQuPFwvcD5cclxuIiwiaW5wdXQiOiI8cD5UaGUgZmlyc3QgbGluZSBvZiBpbnB1dCBjb250YWlucyBvbmUgaW50ZWdlciBzcGVjaWZ5aW5nIHRoZSBudW1iZXIgb2YgdGVzdCBjYXNlcyB0byBmb2xsb3cuIEVhY2ggdGVzdCBjYXNlIGJlZ2lucyB3aXRoIGEgbGluZSBjb250YWluaW5nIGFuIGludGVnZXIgRiwgdGhlIG51bWJlciBvZiBmcmllbmRzaGlwcyBmb3JtZWQsIHdoaWNoIGlzIG5vIG1vcmUgdGhhbiAxMDAgMDAwLiBFYWNoIG9mIHRoZSBmb2xsb3dpbmcgRiBsaW5lcyBjb250YWlucyB0aGUgbmFtZXMgb2YgdHdvIHBlb3BsZSB3aG8gaGF2ZSBqdXN0IGJlY29tZSBmcmllbmRzLCBzZXBhcmF0ZWQgYnkgYSBzcGFjZS4gQSBuYW1lIGlzIGEgc3RyaW5nIG9mIDEgdG8gMjAgbGV0dGVycyAodXBwZXJjYXNlIG9yIGxvd2VyY2FzZSkuPFwvcD5cclxuIiwib3V0cHV0IjoiPHA+V2hlbmV2ZXIgYSBmcmllbmRzaGlwIGlzIGZvcm1lZCwgcHJpbnQgYSBsaW5lIGNvbnRhaW5pbmcgb25lIGludGVnZXIsIHRoZSBudW1iZXIgb2YgcGVvcGxlIGluIHRoZSBzb2NpYWwgbmV0d29yayBvZiB0aGUgdHdvIHBlb3BsZSB3aG8gaGF2ZSBqdXN0IGJlY29tZSBmcmllbmRzLjxcL3A+XHJcbiIsImhpbnQiOiIiLCJvcmlnaW5hbCI6IjEiLCJodG1sX3RpdGxlIjoiMCIsInByb2JsZW1fbGFuZ190Y29kZSI6IkVuZ2xpc2gifV0=

접근 방법

- 0. 이름을 key로 하고, 그 이름에 해당하는 인덱스를 value로 가지는 딕셔너리를 만든다. 그리고 최대 F * 2개의 리스트를 두 개 만들어 하나는 친구 관계의 인원수를 세고, 나머지는 친구 관계의 집합을 표현한다.
- 1. 유니온 파인드를 통해 그룹을 나누어 만들고, 부모 노드에 대해 친구 관계의 수를 갱신해주며 그 값을 결과값에 최대값을 갱신하며 저장한다.

코드

# https://www.acmicpc.net/problem/4195
# 접근 방법
# 0. 이름을 key로 하고, 그 이름에 해당하는 인덱스를 value로 가지는 딕셔너리를 만든다. 그리고 최대 F * 2개의 리스트를 두 개 만들어 하나는 친구 관계의 인원수를 세고, 나머지는 친구 관계의 집합을 표현한다. 
# 1. 유니온 파인드를 통해 그룹을 나누어 만들고, 부모 노드에 대해 친구 관계의 수를 갱신해주며 그 값을 결과값에 최대값을 갱신하며 저장한다.
def get_parent(idx):
    if parent[idx] == idx:
        return idx
    index = get_parent(parent[idx])
    parent[idx] = index
    return index

def find_union(p1, p2):
    x1 = get_parent(p1)
    x2 = get_parent(p2)
    if x1 < x2:
        parent[x2] = x1
        total[x1] += total[x2]
        total[x2] = 0
        return total[x1]    
    elif x1 > x2:
        parent[x1] = x2
        total[x2] += total[x1]
        total[x1] = 0
        return total[x2]
    return total[x1]

import sys
input = sys.stdin.readline
for tc in range(int(input())):
    f = int(input())
    parent = [i for i in range(f*2)]
    total = [1 for i in range(f*2)]
    name_set = set()
    name_dict = {}
    idx = 0
    for _ in range(f):
        p1, p2 = input().split()
        if p1 not in name_set:
            name_dict[p1] = idx
            idx += 1
            name_set.add(p1)
        if p2 not in name_set:
            name_dict[p2] = idx
            idx += 1
            name_set.add(p2)
        count = find_union(name_dict[p1], name_dict[p2])
        print(count)