[Python] 결혼식 (백준 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

힌트

예제 1의 경우 2와 3은 상근이의 친구이다. 또, 3과 4는 친구이기 때문에, 4는 상근이의 친구의 친구이다. 5와 6은 친구도 아니고, 친구의 친구도 아니다. 따라서 2, 3, 4 3명의 친구를 결혼식에 초대한다.

문제 이해

- 그래프 이론을 응용한 문제입니다.

- 입력부분은 일반적인 인접리스트 형태로 그래프를 입력받는 것과 동일하나, BFS/DFS 함수 부분이 일반적인 경우와 다릅니다.

- 탐색의 깊이를 체크하며, 일정 깊이(친구의 친구 보다 커짐)를 초과하면 그를 포함하지 않아야 합니다.

- count 변수를 통해 깊이를 체크하였으며, 한번 노드의 요소들을 탐색할 때마다 count에 +1을 하고, count가 총 2회가 되면 종료합니다.

 

코드

#5567 결혼식

from collections import deque

def BFS(v,graph,visited):

    q = deque()

    visited[v]=True

    q.append(v)

    count = 0

    result = 0

    while q:

        count += 1

        for _ in range(len(q)):

            v = q.popleft()

            for i in range(len(graph[v])):

                if visited[graph[v][i]]==False:

                    visited[graph[v][i]] = True

                    result += 1

                    q.append(graph[v][i])

        if count == 2:

            break

    return result

n=int(input())

m=int(input())

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

for i in range(m):

    a,b = map(int,input().split())

    graph[a].append(b)

    graph[b].append(a)

   

visited =[False]*(n+1)

print(BFS(1,graph,visited))