https://www.acmicpc.net/problem/1260
1260번: DFS와 BFS
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사
www.acmicpc.net
from collections import deque
import sys
input=sys.stdin.readline
n, m, v = map(int, input().split())
visited=[0] * (n+1)
graph=[[0]*(n+1) for _ in range(n+1)]
for _ in range(m):
a, b=map(int, input().split())
graph[a][b]=1
graph[b][a]=1
def dfs(v):
visited[v]=1
print(v, end=' ')
for i in range(1, n+1):
if visited[i]==0 and graph[v][i]==1:
dfs(i)
def bfs(v):
queue=deque([v])
visited[v]=0
while queue:
v=queue.popleft()
print(v, end=' ')
for i in range(1, n+1):
if visited[i]==1 and graph[v][i]==1:
queue.append(i)
visited[i]=0
dfs(v)
print()
bfs(v)
'알고리즘 > 그래프' 카테고리의 다른 글
[백준/그래프] 13023 ABCDE (Python, 파이썬) (0) | 2021.06.19 |
---|