[Algorithm] [Python & Swift] 백준/BOJ - 1260 _ DFS와 BFS

1260 - DFS와 BFS

---

INDEX

• INDEX
  • ### 문제 설명
  • ### 입력
  • ### 출력
  • ### 예제 입출력
  • ### 제출 코드
  • ### 코드 설명

문제 설명

---

그래프를 DFS로 탐색한 결과와 BFS로 탐색한 결과를 출력하는 프로그램을 작성하시오. 단, 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문하고, 더 이상 방문할 수 있는 점이 없는 경우 종료한다. 정점 번호는 1번부터 N번까지이다.

입력

---

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사이에 여러 개의 간선이 있을 수 있다. 입력으로 주어지는 간선은 양방향이다.

출력

---

첫째 줄에 DFS를 수행한 결과를, 그 다음 줄에는 BFS를 수행한 결과를 출력한다. V부터 방문된 점을 순서대로 출력하면 된다.

예제 입출력

---

입력 :

4 5 1

1 2

1 3

1 4

2 4

3 4

출력 :

1 2 4 3

1 2 3 4

제출 코드

---

Python

from collections import deque
from sys import stdin

def DFS(graph, start):
    visited = []
    stack = [start]

    while stack:
        node = stack.pop()
        if node not in visited:
            visited.append(node)
            stack.extend(sorted(graph[node], reverse=True))         # LIFO , 역정렬해서 넣어줘야 작은 수 부터 나옴.

    return visited

def BFS(graph, start):
    visited = []
    queue = deque([start])

    while queue:
        node = queue.popleft()
        if node not in visited:
            visited.append(node)
            queue.extend(sorted(graph[node]))                      # FIFO, 정렬해서 넣어줘야 작은 수 부터 나옴.

    return visited

# N : node, M : Edge, V : start Node
N, M, V = map(int, stdin.readline().split())

graph = [set([]) for _ in range(N+1)]

for _ in range(M):
    a,b = map(int, stdin.readline().split())
    graph[a].add(b)
    graph[b].add(a)

print(*DFS(graph, V))
print(*BFS(graph, V))

Swift

import Foundation

func DFS(start: Int) {
    var visited = [Int]()
    var stack = [start]

    while !stack.isEmpty {
        let node = stack.popLast()!

        if !visited.contains(node) {
            visited.append(node)
            print(node, terminator: " ")
            stack.append(contentsOf: graph[node].sorted(by: >))
        }
    }
}

func BFS(start: Int) {
    var visited = [Int]()
    var queue = [start]

    while !queue.isEmpty {
        let node = queue.removeFirst()

        if !visited.contains(node) {
            visited.append(node)
            print(node, terminator: " ")
            queue.append(contentsOf: graph[node].sorted())
        }
    }
}

var readline = readLine()!.split(separator: " ").map { Int($0)! }
let n = readline[0]
let m = readline[1]
let v = readline[2]
var graph: [[Int]] = Array(repeating: Array(repeating: 0, count: 0), count: n + 1)

for _ in 0..<m {
    let readline = readLine()!.split(separator: " ").map { Int($0)! }
    let src = readline[0]
    let dest = readline[1]
    graph[src].append(dest)
    graph[dest].append(src)
}

DFS(start: v)
print()
BFS(start: v)

코드 설명

---

해당 문제에서 주의깊게 살펴볼 점은 아래와 같다.

• 양방향 그래프라는 것이다.
  즉, 1에서 2가 이어지면 2에서 1 또한 이어진다.

• 현재 노드에 연결된 노드가 여러개라면, 작은 수 부터 출력될 수 있도록 정렬 조절이 필요하다.

• 양방향 그래프기 때문에 중복 제거를 위해 set()을 이용한다.

DFS 함수는 list를 이용하며 리스트에 현재 정점에 연결된 정점들을 연결해줄때 LIFO 성질을 이용하여 작은수 부터 출력시키기 위해 내림차순 정렬하여 넣어준다.

BFS 함수는 deque를 이용하여 덱에 현재 정점에 연결된 정점들을 연결해줄때 FIFO 성질을 이용하여 작은수 부터 출력시키기 위해 오름차순 정렬하여 넣어준다.