[구현/수학] 백준 26069 붙임성 좋은 총총이 - 파이썬(Python)

 

[ Contents ]

 

 

1. 문제 (링크 참조)

26069번: 붙임성 좋은 총총이

 

 

2. 문제 풀이

 ChongChong과 만난 사람들은 모두 춤을 추게 되버리는 무서운 문제입니다. 마지막까지 춤을 추고 있는 사람을 구해야 합니다.

 

 

3. 코드

import sys
input = sys.stdin.readline

n = int(input())
dance = {'ChongChong'}

for i in range(1, n+1):
    a, b = input().rstrip().split()

    if a in dance:
        dance.add(b)

    if b in dance:
        dance.add(a)

print(len(dance))

 파이썬의 집합 Set 자료구조를 이용합니다. 춤추는 사람의 모임을 dance에 넣고, 그 상대편도 dance에 넣습니다.