[BOJ] 26069번 붙임성 좋은 총총이

[BOJ] 26069번 붙임성 좋은 총총이 문제

무난한 해시 문제다. "무지개 댄스"에 꽃히기 쉽지만 최대한 문제를 자세히 읽어보자.

핵심은 매우 간단하다. 주어지는 문자열 중에서 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))

 

딱히 어려운 문제는 아니기에 로직을 짜는데에 있어서 어려움은 없다.