What I Learned/Algorithm Practice

[백준 - python] 26069번: 붙임성 좋은 총총이

Interrobang 2023. 7. 28. 19:30

문제 링크

 

26069번: 붙임성 좋은 총총이

첫번째 줄에는 사람들이 만난 기록의 수 $N\ (1 \le N \le 1\ 000)$이 주어진다. 두번째 줄부터 $N$개의 줄에 걸쳐 사람들이 만난 기록이 주어진다. $i + 1$번째 줄에는 $i$번째로 만난 사람들의 이름 $A_i$

www.acmicpc.net

문제 풀이

from sys import stdin

input = stdin.readline

n = int(input())
chong = set(['ChongChong'])

for _ in range(n):
    a, b = input().split()
    if a in chong or b in chong:
        chong.update([a, b])

print(len(chong))

26069 입출력 예시

*key point: 춤을 추고 있는 사람이 담긴 집합 chong을 만들어 활용한다. 입력값 중 이미 집합에 속한 사람이 있을때 모두 집합에 추가하면 쉽게 춤을 추고 있는 사람을 얻을 수 있다.