5567 결혼식

정민용·2023년 2월 7일

백준

목록 보기
16/286

문제

상근이는 자신의 결혼식에 학교 동기 중 자신의 친구와 친구의 친구를 초대하기로 했다. 상근이의 동기는 모두 N명이고, 이 학생들의 학번은 모두 1부터 N까지이다. 상근이의 학번은 1이다.

상근이는 동기들의 친구 관계를 모두 조사한 리스트를 가지고 있다. 이 리스트를 바탕으로 결혼식에 초대할 사람의 수를 구하는 프로그램을 작성하시오.

n = int(input())
m = int(input())

graph = []
for _ in range(n + 1):
  graph.append([])

for _ in range(m):
  a, b = map(int, input().split())
  graph[a].append(b)
  graph[b].append(a)

friends = []

for i in graph[1]:
  if not i in friends:
    friends.append(i)
  for j in graph[i]:
    if not j in friends:
      friends.append(j)

if 1 in friends:
  friends.remove(1)

my_friend = len(friends)
print(my_friend)

백준 5567 결혼식

0개의 댓글