Python/백준 문제 풀이
[BOJ] 26069번 붙임성 좋은 총총이
서원두
2022. 12. 31. 16:49

무난한 해시 문제다. "무지개 댄스"에 꽃히기 쉽지만 최대한 문제를 자세히 읽어보자.
핵심은 매우 간단하다. 주어지는 문자열 중에서 ChongChong이 나온 그 순간부터 모든 문자열을 해시 안에 집어넣으면 된다. 또한 ChongChong은 반드시 세어줘야한다는 것을 잊지 말자. 문제에 ChongChong은 편의상 사람으로 분류한다고 엄연히 명시되어있다.
import sys
if __name__ == '__main__':
rainbow = set()
for N in range(int(sys.stdin.readline().rstrip())):
A, B = sys.stdin.readline().rstrip().split()
if A == 'ChongChong' or B == 'ChongChong':
rainbow.add(A)
rainbow.add(B)
if A in rainbow or B in rainbow:
rainbow.add(A)
rainbow.add(B)
print(len(rainbow))
딱히 어려운 문제는 아니기에 로직을 짜는데에 있어서 어려움은 없다.
728x90