[백준 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도 친구관계이다.

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

solution.py

import sys from collections import defaultdict, deque def bfs(start): cnt = 0 # 각 노드가 방문된 정보를 표현 visited = [0 for _ in range(n+1)] # 현재 노드를 방문 처리 visited[start] = 1 queue = deque([[start, 0]]) # 큐가 빌때까지 반복 while queue: # 친구와 거리 u, dist = queue.popleft() if dist <= 2: # 친구의 친구(dist==2)까지 초대할 수 있다 cnt += 1 for v in g[u]: if not visited[v]: visited[v] = 1 queue.append([v, dist+1]) return cnt-1 # cnt 자기자신 -1 n = int(sys.stdin.readline()) m = int(sys.stdin.readline()) g = defaultdict(list) for _ in range(m): a, b = map(int, sys.stdin.readline().split()) g[a].append(b) g[b].append(a) print(bfs(1))

출처

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

from http://kangsu-2ji.tistory.com/54 by ccl(A) rewrite - 2021-11-17 00:26:24