Programmers: 네트워크

문제 독해를 잘못해서 헷갈렸던 문제. 여타 bfs/dfs 문제랑 비슷하게 푸는데, 바깥에서 모든 컴퓨터를 순회하면서 한 번이라도 방문한 곳은 visited 변수에 담아 체크했다. 이미 방문한 곳은 컴퓨터끼리 연결되었다는 뜻이니까 ㅇㅇ

문제

https://programmers.co.kr/learn/courses/30/lessons/43162

내가 작성한 코드

Swift

import Foundation

func solution(_ n:Int, _ computers:[[Int]]) -> Int {
    var graph = [Int: [Int]]()
    for i in 0..<n {
        graph[i] = []
    }
    for i in 0..<n {
        for j in 0..<n {
            if i == j {
                continue
            }
            if computers[i][j] == 1 {
                graph[i, default: []].append(j)
            }
        }
    }

    func bfs(_ start: Int) -> Set<Int> {
        var visited: Set<Int> = []
        var queue = [start]
        while !queue.isEmpty {
            let v = queue.removeFirst()
            for w in graph[v]! {
                if !visited.contains(w) {
                    visited.insert(w)
                    queue.append(w)
                }
            }
        }
        return visited
    }

    var visited: Set<Int> = []
    var answer = 0
    for i in 0..<n {
        if !visited.contains(i) {
            let v = bfs(i)
            for j in v {
                visited.insert(j)
            }
            answer += 1
        }
    }
    return answer
}