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))

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