백준 / 1260번 / DFS와 BFS / DFS, BFS / python 파이썬

 

문제 : https://www.acmicpc.net/problem/1260

 

 

코드 

from collections import deque
import sys

input = sys.stdin.readline
n, m, v = map(int, input().split())
graph = [[] for _ in range(n+1)]

for _ in range(m):
    a, b = map(int, input().split())
    graph[a].append(b)
    graph[b].append(a)

def DFS(start):
    visited[start] = True
    print(start, end = " ")
    for i in graph[start]:
        if not visited[i]:
            DFS(i)

def BFS(start):
    queue = deque([start])
    visited[start] = True
    while queue:
        p = queue.popleft()
        print(p, end = " ")
        for i in graph[p]:
            if not visited[i]:
                visited[i] = True
                queue.append(i)

for i in graph:
    i.sort()

visited = [False] * (n + 1)

DFS(v)
print()

visited = [False] * (n + 1)
BFS(v)

 

 

알아둘 것

for i in graph:
    i.sort()

 

" 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. " 조건에 의해 오름차순으로 정렬한 후, 탐색을 진행해야 원하는 결과를 얻을 수 있다.