코딩테스트/백준
[백준] 1260번 DFS와 BFS / 파이썬(python)
Jiwon_C
2021. 5. 17. 23:53
# 문제 링크
https://www.acmicpc.net/problem/1260
1260번: DFS와 BFS
첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사
www.acmicpc.net
# Soultion
dfs와 bfs에 대해 알면 쉽게 풀 수 있는 문제이다.
주의) 인접한 노드가 작은 순으로 방문한다고 했으므로, 13,28번째 줄처럼 sorted()를 이용하여 오름차순 정렬해준다.
|
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
29
30
31
32
|
n,m,v = map(int,input().split())
graph = [[] * n for _ in range(n+1)]
for _ in range(m):
a,b = map(int,input().split())
graph[a].append(b)
graph[b].append(a)
visited = [0] * (n+1)
def dfs(v):
print(v,end=' ')
global visited, graph
visited[v] = 1
for i in sorted(graph[v]):
if visited[i]!=1:
dfs(i)
dfs(v)
print()
from collections import deque
visited = [0] * (n+1)
def bfs(start):
global visited, graph
queue = deque([start])
visited[start] = 1
while queue:
v = queue.popleft()
print(v,end=' ')
for i in sorted(graph[v]):
if visited[i]!=1:
queue.append(i)
visited[i] = 1
bfs(v)
|
cs |