[백준] 1260번 DFS와 BFS [Swift]

문제 링크

문제

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

입력

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

출력

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

문제 풀이

DFS는 깊이 우선 탐색, BFS는 너비 우선 탐색으로 해당 알고리즘에 맞게 정점들을 탐색하면 된다. DFS는 스택을 사용하여 탐색하면 되고 BFS는 큐를 사용하여 탐색하면 된다. 단 방문할 수 있는 정점이 여러 개인 경우에는 정점 번호가 작은 것을 먼저 방문해야 하기 때문에 각각의 탐색에 맞는 정렬이 필요하다.

소스코드

import Foundation

let firstLine = readLine()!.split(separator: " ").map(){Int(String($0))!}
let N = firstLine[0]
let M = firstLine[1]
let V = firstLine[2]
var graph = [Int: [Int]]()

for _ in 0..<M {
    let line = readLine()!.split(separator: " ").map(){Int(String($0))!}
    let start = line[0]
    let end = line[1]

    if graph[start] == nil {
        graph[start] = [end]
    } else {
        graph[start]?.append(end)
    }
    if graph[end] == nil {
        graph[end] = [start]
    } else {
        graph[end]?.append(start)
    }
}
for key in graph.keys {
    graph[key]?.sort(by: >)
}
var dfsResult = ""
for i in DFS(graph,V) {
    dfsResult += String(i) + " "
}

for key in graph.keys {
    graph[key]?.sort(by: <)
}
var bfsResult = ""
for i in BFS(graph,V) {
    bfsResult += String(i) + " "
}

print(dfsResult)
print(bfsResult)

func DFS(_ graph: [Int: [Int]], _ start: Int ) -> [Int]{

    var visited = [Int]()
    var stack = [start]

    while stack.count != 0 {
        let node = stack.popLast()!
        if visited.contains(node) {
            continue
        } else {
            visited.append(node)
            if let nodeWay = graph[node] {
                stack += nodeWay
            }
        }
    }
    return visited
}

func BFS(_ graph: [Int: [Int]], _ start: Int) -> [Int]{
    var visited = [Int]()
    var queue = [start]

    while queue.count != 0 {
        let node = queue.remove(at: 0)
        if visited.contains(node){
            continue
        } else {
            visited.append(node)
            if let nodeWay = graph[node] {
                queue += nodeWay
            }
        }
    }
    return visited
}