백준 26069번 python

말이 길지만 결국 "ChongChong" 과 같은 line에서 입력되면 count가 하나 늘어나는 구조 + 해당 입력값도 ChongChong과 똑같은 능력을 가진다는 말이다.

 

풀이:

set을 하나 만들고 입력값 둘 중 하나가 set에 있다면 둘 다 set에 add하고 마지막에 length값을 출력한다.

set은 중복을 허용하지 않으므로 중복된 값을 add해도 상관 없다.

import sys
input = sys.stdin.readline
N = int(input().rstrip())
dancePeople = set(["ChongChong"])
for _ in range(N):
    persons = input().rstrip().split(" ")
    if persons[0] in dancePeople or persons[1] in dancePeople:
        dancePeople.add(persons[0])
        dancePeople.add(persons[1])
print(len(dancePeople))