boj 5567 결혼식 (실버1)

문제

https://www.acmicpc.net/problem/5567

내풀이

import sys
input = sys.stdin.readline

f = int(input())
n = int(input())

d = {}
for _ in range(n):
    a,b = map(int,input().split())

    if a in d:
        d[a].add(b)

    else :
        d[a] = {b}

    if b in d:
        d[b].add(a)

    else :
        d[b] = {a}

answer_list = set()

if 1 in d:  # 상근이가 친구가 있으면
    friends = d[1] #상근이 친구목록 정리
    answer_list.update(d[1])  # 상근이 친구들 리스트에 추가

    for friend in friends:   # 상인이 친구들 탐색하며 친구의 친구 구하기
        if friend in d:  # 친구의 친구가 존재하면
            answer_list.update(d[friend])  #리스트에 추가

    print(len(answer_list)-1)

else :
    print(0)

결과