[Swift] 백준2606 바이러스

import Foundation

/// N : 컴퓨터의 수 (정점)
/// M : 네트워크 상에서 직접 연결되어 있는 컴퓨터쌍의 수 (간선)

let N = Int(readLine()!)!
let M = Int(readLine()!)!

var graph = [String: [String]]()
var cnt = 0

for i in 0..<N {
    graph[String(i+1)] = []
}

for _ in 0..<M{
    let ab = readLine()!.components(separatedBy: " ")
    let a = ab[0]
    let b = ab[1]
    graph[a]?.append(b)
    graph[b]?.append(a)
}

bfs("1")
print(cnt - 1)

func bfs(_ start: String) -> [String] {
    var visitedQueue: [String] = []
    var needVisitQueue: [String] = [start]
    
    while !needVisitQueue.isEmpty {
        let node: String = needVisitQueue.removeFirst()
        if visitedQueue.contains(node) { continue }
        
        visitedQueue.append(node)
        cnt += 1
        needVisitQueue += graph[node] ?? []
    }
    
    return visitedQueue
}

bfs로 방문하면서 count하도록 간단하게 풀었다.