친구 네트워크

주말에 갑자기 c++ 로 코테를 치게 되어서 3일동안 c++을 공부했다

예전 자료구조 수업을 c로 들었던 것 그리고 컴프실을 c로 들었던가

특히 ~자동으로 알아서 다 해주는~ 파이썬을 쓰다가 c++ 쓰려니까 수동기어가 달린 차를 모는 것 같은 느낌이다

언어가 c, c++, java밖에 없던데 c로 필요한 자료구조 다 만들어서 코테를 통과하려면 피지컬이 대단해야 할 것 같다

암튼 타입 지정, 세미콜론, vector, 필요한 파라미터 전달같은 것에 익숙해져 있다가 다시 파이썬으로 오니까 또 관성대로 하게 된다

 

친구 네트워크

문제

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

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

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

입력

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

출력

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

저 두 사람의 친구 네트워크에 몇 명이 있냐는 말이 이해가 안 돼서 한참 시간을 썼다

국어공부를 더 해야할까 ?? (ㅠㅠ)

 

처음에는 dfs로 짰다가, 시간초과가 나서 union find 를 이용해서 풀었다

친구 수를 어떻게 저장해야 할지 몰라서 p 배열에서 이리저리 돌려보고 했는데 그냥 따로 공간을 내주면 간단하게 해결된다

 

import sys

input = sys.stdin.readline
tk = int(input())


def get_parent(x):
    if x != p[x]:
        p[x] = get_parent(p[x])
    return p[x]


def union(a, b):
    x = get_parent(a)
    y = get_parent(b)
    if x != y:
        if x > y:
            p[x] = y
            friend_count[y] += friend_count[x]
        else:
            p[y] = x
            friend_count[x] += friend_count[y]


for _ in range(tk):
    friend = {}
    f = int(input())
    p = dict()
    friend_count = dict()
    friendName = set()
    for _ in range(f):
        a, b = map(str, input().split())
        if a not in p:
            p[a] = a
            friend_count[a] = 1
        if b not in p:
            p[b] = b
            friend_count[b] = 1
        union(a, b)
        print(max(friend_count[p[a]], friend_count[p[b]]))