[Baekjoon]백준 26069 붙임성 좋은 총총이(실버 4) - Python

문제설명 1

문제설명 2

문제를 살펴보면 총총이를 만난 순간부터 춤을 추게 되며 총총이가 아니라도 춤추는 누군가를 보게 되면 함께 춤을 춘다

저번 문제에서 집합을 사용하지 않았는데, 이번에는 집합을 사용하여 코드를 작성하였다

일단 N을 입력받고 for문을 통해 반복문으로 이름을 받아준다

이후에 in 연산자를 통해 춤을 추고 있는 사람인지 아닌지 확인한 후에 만약 한명이라도 춤을 추고 있다면

add 연산을 통하여 집합에 이름을 넣어준다 

이후에는 집합의 길이를 출력해주면 된다

N = int(input())
dance = set()
dance.add('ChongChong')

for _ in range(N):
    person = input().split()
    if person[0] in dance:
        dance.add(person[1])
    elif person[1] in dance:
        dance.add(person[0])
        
print(len(dance))