반응형
https://www.acmicpc.net/problem/24445
24445번: 알고리즘 수업 - 너비 우선 탐색 2
첫째 줄에 정점의 수 N (5 ≤ N ≤ 100,000), 간선의 수 M (1 ≤ M ≤ 200,000), 시작 정점 R (1 ≤ R ≤ N)이 주어진다. 다음 M개 줄에 간선 정보 u v가 주어지며 정점 u와 정점 v의 가중치 1인 양
www.acmicpc.net
from collections import deque
import sys
sys.setrecursionlimit(10 ** 9)
input1 = sys.stdin.readline
# 차례대로 정점의 수 , 간선의 수 , 시작 정점.
n, m, r = map(int, input1().split())
graph = [[] for _ in range(n + 1)]
for _ in range(m):
a, b = map(int, input1().split())
graph[a].append(b)
graph[b].append(a)
visited = [0] * (n + 1)
cnt = 1
def bfs(start):
global cnt
queue = deque([r])
visited[r] = 1
while queue:
v = queue.popleft()
graph[v].sort(reverse=True)
for i in graph[v]:
if visited[i] == 0:
cnt +=1
queue.append(i)
visited[i] += cnt
bfs(r)
for i in range(1, n + 1):
print(visited[i])
반응형
'알고리즘 풀이 > DFS & BFS' 카테고리의 다른 글
[백준/Python] 1260번 : DFS와 BFS (0) | 2022.12.29 |
---|---|
[백준 / Python] DFS_2606번: 바이러스 (0) | 2022.12.29 |
[BFS] 백준 24444번 (0) | 2022.12.27 |
BFS_미로탈출 이코테 (0) | 2022.12.27 |
DFS_ 음료수 얼려 먹기 이코테 (0) | 2022.12.27 |
댓글