[Swift] 2606번: 바이러스

 

  문제  

2606번: 바이러스

---

  알고리즘  

DFS 사용

1. DFS를 수행한다.(시작노드 = 1)
2. 방문한 노드의 개수를 구한 후 1을 빼서 출력한다.
    ( 1에 의해 바이러스에 걸린 컴퓨터의 개수 => 1은 포함하지 않음)

---

  나의 코드  

// dfs
func dfs(n: Int){
    visited[n] = true
    for i in graph[n]{
        if !visited[i]{
            dfs(n: i)
        }
    }
}

// 선언 및 초기화
let node: Int = Int(readLine()!)!
let edge: Int = Int(readLine()!)!
var graph = [[Int]].init(repeating: [], count: node + 1)
var visited = [Bool].init(repeating: false, count: node+1)

// 그래프 입력받기
for _ in 1...edge{
    let info = readLine()!.split(separator: " ").map{ Int(String($0))! }
    graph[info[0]].append(info[1])
    graph[info[1]].append(info[0])
}

// dfs 수행(시작노드 = 1)
dfs(n: 1)

// 1에 의해 감염된 컴퓨터의 개수
// 방문한 노드의 개수를 구한 후 1을 뺌
print( visited.filter{($0) == true}.count - 1 )

---

  Key point!  

📍DFS / BFS 개념 및 코드

[Algorithm] DFS / BFS 탐색 알고리즘

---