인프런 커뮤니티 질문&답변

안녕하세요 강사님 새해 복 많이 받으세요~

1012번 문제는 인접행렬로 풀으셨는데 인접리스트로 푸는 방법은 없을까요?

"인접행렬로는 앞 문제에서 풀어봤으니까 인접리스트로 풀어봐야지!" 해서 풀어보고 있는데 dfs가 안되는것 같아서 질문드려봅니다!

import Foundation

var testCase = Int(readLine()!)!

var visited: [Bool] = []
var adjList: [[Int]] = [[]]

for _ in 0..<testCase {
    let info = readLine()!.split(separator: " ").map { Int($0)! }
    let M = info[0] // 주의! 보통 세로 가로이지만 여기서는 가로 세로이다.
    let N = info[1]
    let K = info[2]
    print(search(N,M,K))
}


func search(_ N: Int, _ M: Int, _ K: Int) -> Int {
    var result = 0
    adjList = [[Int]](repeating: [], count: M)
    visited = [Bool](repeating: false, count: M)
    
    for _ in 0..<K {
        let point = readLine()!.split(separator: " ").map { Int($0)! }
        let x = point[0]
        let y = point[1]
        adjList[y].append(x)
    }
    
    // MARK: - 지금까지는 데이터 다 들어감 이제 dfs돌리기
    for i in 0..<N {
        if visited[i] == false {
            print("now: \(i)")
            result += 1
            dfs(i)
        }
    }

    return result
}

func dfs(_ node: Int) {
    visited[node] = true
    for i in 0..<adjList[node].count {
        if visited[adjList[node][i]] == false {
            dfs(adjList[node][i])
        }
    }
}

언어는 swift인데 많이 비슷해서 읽으시는데는 문제 없으실겁니다!