[맛있는물회] [파이썬] <백준 알고리즘> 4195번 "친구 네트워크"

문제 조건

---

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

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

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

 

Input

---

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

 

Output

---

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

 

생각한 아이디어

---

흡,,

 

Union - find 는 딕셔너리를 이용해서 쉽게 구현했다.

맨첨에 count하는 부분을

def count(x):
   root = find(x)
   cnt=0
   for name in names:
      if find(name)==root:
         cnt+=1
   return cnt

 

이렇게 아주 시간초과 나게끔 모든 노드의 parent를 찾고 만약 그 parent가 내가 찾는 것이라면 증가하고 이렇게 했다

그랬더니 시간초과가 발생했다.

 

조금 생각을 다르게 해서 count 하는 리스트를 만들었다.

그리고 count를 union하는 부분에서 해주었다.

그런데 계속 메모리초과가 발생하는 것이었다..

진짜 계속 고생했다 ㅜㅜ

근데 추가해주는 부분이 만약 x,y의 parent가 같을때는 추가해주면 안되는데 계속 추가해줘서 그런것 같다.

(근데 왜 이게 틀렸습니다가 아니라, 메모리초과 일까?)

(흠,, 100000만명의 친구 구성이 들어왔을때 같은 친구 구성이 계속 들어온다면? count 값이 너무 커져서 메모리 초과가 발생하는 건가? ) 

어쨌든 저것때문에 계속 고생하다가 저부분만 고쳐주니 통과되었다 ㅜㅜ

 

소스코드

#4195 친구 네트워크

import sys
input=sys.stdin.readline

def find(x):
   if parent[x]==x:
      return x
   parent[x]=find(parent[x])
   return parent[x]

def union(x,y):
   x,y=find(x),find(y)
   parent[x]=y
   if x!=y:
       cnt_parent[y].append(x)
       cnt_parent[y].extend(cnt_parent[x])
       cnt_parent[x]=[]
   

for _ in range(int(input())):
   parent=dict()
   cnt_parent=dict()
   for _ in range(int(input())):
      a=list(map(str,input().split()))
      if a[0] not in parent:
         cnt_parent[a[0]]=[]
         parent[a[0]]=a[0]
      if a[1] not in parent:
         cnt_parent[a[1]]=[]
         parent[a[1]]=a[1]

      union(a[0],a[1])
      print(len(cnt_parent[find(a[0])])+1)

 

*파이썬 문법 정리