[Algorithm] DFS와 BFS

[hwangJi-dev]

💬 문제

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

💬 Idea

• DFS를 수행할 때 번호가 작은 노드들부터 순회하기 위해서 순회 배열을 정렬한 뒤 뒤집어서 저장해준다.
• BFS를 수행할 때도 번호가 작은 노드들부터 순회하기 위해서 순회 배열을 정렬하여 저장해준다.

💬 풀이

func solution1260() {
 let info = readLine()!.split(separator: " ").map({ Int(String($0))! })
 var graph: [Int: [Int]] = [:]

 for _ in 0..<info[1] {
 let nodes = readLine()!.split(separator: " ").map({ Int(String($0))! })
 
 if graph[nodes[0]] == nil {
 graph[nodes[0]] = [nodes[1]]
 } else {
 graph[nodes[0]]?.append(nodes[1])
 }
 
 if graph[nodes[1]] == nil {
 graph[nodes[1]] = [nodes[0]]
 } else {
 graph[nodes[1]]?.append(nodes[0])
 }
 }

 func dfs(graph: [Int: [Int]], start: Int) -> [Int] {
 var needToVisitStack: [Int] = [start]
 var visitedQueue: [Int] = []
 
 while !needToVisitStack.isEmpty {
 let node = needToVisitStack.removeLast()
 
 if visitedQueue.contains(node) { continue }
 
 visitedQueue.append(node)
 if let nodes = graph[node]?.sorted().reversed() {
 needToVisitStack += nodes
 }
 }
 
 return visitedQueue
 }

 func bfs(graph: [Int: [Int]], start: Int) -> [Int] {
 var needToVisitQueue: [Int] = [start]
 var visitedQueue: [Int] = []
 
 while !needToVisitQueue.isEmpty {
 let node = needToVisitQueue.removeFirst()
 
 if visitedQueue.contains(node) { continue }
 
 visitedQueue.append(node)
 if let nodes = graph[node]?.sorted() {
 needToVisitQueue += nodes
 }
 }
 
 return visitedQueue
 }

 print(dfs(graph: graph, start: info[2]).map({ String($0) }).joined(separator: " "))
 print(bfs(graph: graph, start: info[2]).map({ String($0) }).joined(separator: " "))
}

소요시간 : 30분