CS/알고리즘_문제풀이(파이썬)

깊이우선탐색과 너비우선탐색

Jedy_Kim 2021. 7. 26. 15:41
728x90

문제

그래프가 주어질 때, 0번 정점을 시작으로 하여 깊이우선탐색과 너비우선탐색의 결과를 출력하는 프로그램을 작성하시오. 단, 노드를 방문할 때는 노드 번호가 작은 순서대로 방문한다고 하자.

 

입력

첫째 줄에 정점의 개수 N과 간선의 개수 M이 주어진다. (1 ≤ N ≤ 1,000, 1 ≤ M ≤ 100,000 ) 둘째 줄부터 간선의 정보가 주어진다. 각 줄은 두 개의 숫자 a, b로 이루어져 있으며, 이는 정점 a와 정점 b가 연결되어 있다는 의미이다. 정점의 번호는 0번부터 N-1번까지이다.

 

출력

첫 번째 줄에 깊이우선탐색 결과, 두 번째 줄에 너비우선탐색 결과를 출력한다.

 

예제 입력

9 12
0 1
0 2
0 3
1 5
2 5
3 4
4 5
5 6
5 7
5 8
6 7
7 8

예제 출력

0 1 5 2 4 3 6 7 8
0 1 2 3 5 4 6 7 8

 

#코드

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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
import sys
from collections import deque
 
def DFS(x):
  global n, myGraph, visited
   
  visited[x] = True
  print(x, end=' ')
  myGraph[x].sort()
  for i in range(len(myGraph[x])):
    x2 = myGraph[x][i]
    if not visited[x2]: 
      DFS(x2) 
 
def BFS():
  global n, myGraph
  visited=[False]*(n+1)
  
  deq = deque()
  deq.append(0)
  visited[0]= True
  
  while deq:
    cur = deq.popleft()
    print(cur, end=' ')
    myGraph[cur].sort()
    for i in range(len(myGraph[cur])):
      nex = myGraph[cur][i]
      if not visited[nex]:
        visited[nex]=True
        deq.append(nex)
        
if __name__ == "__main__":
  input = sys.stdin.readline
  n, m    = map(int, input().split())
  myGraph = [[] for _ in range(n+1)]
  visited = [False]*(n+1
 
  for i in range(m):
    a, b = map(int, input().split())
    myGraph[a].append(b)
    myGraph[b].append(a)  
  
 
  DFS(0
  print()
  BFS()
cs

 

반응형

'CS > 알고리즘_문제풀이(파이썬)' 카테고리의 다른 글

2020 KAKAO BLIND RECRUITMENT [문자열 압축]  (0) 2021.07.29
이상한 계산기  (0) 2021.07.28
점프  (0) 2021.07.23
DSLR  (0) 2021.07.22
수 이어 쓰기 1  (0) 2021.07.21