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

알고리즘

- bfs 탐색을 통해 문제를 수행한다.

- 자기 자신부터 탐색하여 친구의 친구까지 탐색한다.

- 탐색 과정에서 몇 번째 순서로 방문 탐색하였는지 카운트한다.

- 탐색이 끝나면 방문 순서가 2번째인 사람과 3번째인 사람의 수를 합하여 출력한다.

코드

import sys
from collections import deque


# bfs 탐색
def bfs(x):
    queue = deque([x])
    cnt = 0

    # 자기 자신부터 방문
    visited[x] = 1

    while queue:
        friend = queue.popleft()
        cnt += 1

        # 친구의 친구를 확인
        for j in graph[friend]:
            # 방문하지 않았다면 방문
            if not visited[j]:
                
                # 방문 순서를 카운트
                visited[j] = visited[friend] + 1
                queue.append(j)


n = int(sys.stdin.readline())
m = int(sys.stdin.readline())

# 노드 방문 여부와 순서 확인
visited = [0 for i in range(n + 1)]

# 각 연결된 노드를 표현
graph = [[] for _ in range(n + 1)]
for _ in range(m):
    a, b = map(int, sys.stdin.readline().split())
    graph[a].append(b)
    graph[b].append(a)

# 나부터 탐색
bfs(1)

# 방문 순서가 2번째인 사람과 3번째인 사람의 수를 합하여 출력
print(visited.count(2) + visited.count(3))

github

GitHub - junjange/CodingTest: 내가 푼 코딩 테스트 문제와 해결법