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

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

정답 코드 및 설명

import sys
input = sys.stdin.readline

N = int(input())
dancing = set(["ChongChong"])

for _ in range(N):
    a, b = input().split()
    if a in dancing or b in dancing:
        dancing.add(a)
        dancing.add(b)

print(len(dancing))

무지개 댄스를 추는 사람을 dancing 이라는 set 집합에 넣고 제일 처음에는 총총이만 추고 있기에 총총이만을 넣는다.
총총이와 만나야하기에 입력된 두 이름 중 한명이라고 dancing 에 있다면 dancing 에 둘다 넣는다. set 자료 구조를 사용하기 때문에 a와 b 중 어떤 것이 춤을 추는 지 안추는 지 상관이 없다.