Python3/백준 알고리즘

2022.03.17 [백준] (python 파이썬) 케빈 베이컨의 6단계 법칙

ian's coding 2022. 3. 17. 21:27
728x90
반응형

 

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

 

1389번: 케빈 베이컨의 6단계 법칙

첫째 줄에 유저의 수 N (2 ≤ N ≤ 100)과 친구 관계의 수 M (1 ≤ M ≤ 5,000)이 주어진다. 둘째 줄부터 M개의 줄에는 친구 관계가 주어진다. 친구 관계는 A와 B로 이루어져 있으며, A와 B가 친구라는 뜻

www.acmicpc.net


풀이

이 문제는 그래프를 이용해 각 번호에서 갈 수 있는 번호를 저장했다.

이 후, 각 번호에서의 케빈 베이컨의 수가 최저인지 확인 하기 위해 min과 최저이면 어떤 번호에서 최저인지 확인하기 위해 min_p를 선언했다.

그리고 for문을 돌려 각 번호에서의 케빈 베이컨 수를 bfs를 이용해서 구했다. 그리고 이전의 케빈 베이컨 수와 비교하여 작으면 min과 min_p를 업데이트해주었다.

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from collections import deque
 
n,m=map(int, input().split())
graph=[[] for i in range(n+1)]
 
for i in range(m):
    x,y=map(int,input().split())
    graph[x].append(y)
    graph[y].append(x)
 
min=10**6
min_p=0
 
for i in range(1,n+1):
    vis = [-1* (n + 1)
    deq = deque()
    deq.append(i)
    vis[i]=0
    while deq:
        x=deq.popleft()
        for j in graph[x]:
            if vis[j]==-1:
                vis[j]=vis[x]+1
                deq.append(j)
    if sum(vis)<min:
        min=sum(vis)
        min_p=i
print(min_p)
cs

 

 

 

728x90
반응형