[Pyton] 백준 26069 붙임성 좋은 총총이

문제 26069

먼저 위 문제를 이해해보자.

• 첫번째 줄에는 사람들이 만난 기록의 수 n이 주어진ㄴ다.
• 다음줄 부터는 n개의 기록이 주어지며 각 기록마다 만난 사람의 이름이 주어진다.
• 맨 처음 춤을 추던 사람은 'ChongChong'밖에 없었다. 이후 사람들이 만날 때 자신과 만난 사람이 춤을 추는 사람이면 자신도 같이 춤을 춘다.
• 이렇게 총 n번의 만남이 끝난 후에 춤을 추고있는 사람의 수를 출력하는 문제이다.

# 26069
import sys

n = int(sys.stdin.readline().rstrip())
lst = set()
lst.add('ChongChong')
for _ in range(n):
    a, b = map(str, sys.stdin.readline().split())
    if a in lst:
        lst.add(b)
    elif b in lst:
        lst.add(a)

print(len(lst))

위 문제는 set()을 활용하여 한번에 해결했다. 만약 list를 사용하였다면 시간초과가 발생했을 것 같다.