[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분