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
반응형