[백준] 결혼식

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

5567번: 결혼식

 

dictionary 를 이용하여 친구 관계를 나타내고 1 과 dic[1] (1 의 친구들) 의 친구들을 friends 집합에 담아서 중복을 없앤다.

 

friends 에 원소가 있는 경우 1 이 존재하므로 len(friens) - 1 을 반환해주고 그렇지 않으면, 0 을 반환해준다. 

 

import sys
from collections import defaultdict

n = int(sys.stdin.readline())
m = int(sys.stdin.readline())
dic = defaultdict(list)
friends = set()

for i in range(m):
    a, b = map(int, sys.stdin.readline().split())
    dic[a].append(b)
    dic[b].append(a)

friends.update(dic[1])
    
for i in dic[1]:
    friends.update(dic[i])

print(len(friends) - 1 if friends else 0)