[Baekjoon] 결혼식 / Python, 파이썬 / 5567

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

5567번: 결혼식

문제 설명

문제

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

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

입력

첫째 줄에 상근이의 동기의 수 n (2 ≤ n ≤ 500)이 주어진다. 둘째 줄에는 리스트의 길이 m (1 ≤ m ≤ 10000)이 주어진다. 다음 줄부터 m개 줄에는 친구 관계 ai bi가 주어진다. (1 ≤ ai < bi ≤ n) ai와 bi가 친구라는 뜻이며, bi와 ai도 친구관계이다. 

출력

첫째 줄에 상근이의 결혼식에 초대하는 동기의 수를 출력한다.

예제 입력 1

6
5
1 2
1 3
3 4
2 3
4 5

예제 출력 1

3

예제 입력 2

6
5
2 3
3 4
4 5
5 6
2 5

예제 출력 2

0

문제 풀이

import sys
from collections import deque

input = sys.stdin.readline

def bfs(x):
    q = deque([x])

    while q:
        now = q.popleft()
        for x in graph[now]:
            if distance[x] == -1:
                distance[x] = distance[now] + 1
                q.append(x)

N = int(input())
M = int(input())

graph = [[] for _ in range(N+1)]

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

distance = [-1] * (N + 1)
distance[1] = 0

bfs(1)

result = 0
for i in range(1, N + 1):
    if distance[i] != -1 and distance[i] <= 2:
        result += 1

print(result - 1)