[프로그래머스] 무인도 여행

문제

프로그래머스

입력

 maps:[String] = 지도 문자열
 
 maps[i]
 - "X" = 바다
 - 숫자 = 무인도에 머물 수 있는 수

결과

ans: [Int] = 각 섬마다 최대 머물 수 있는 일 수 , 하나 없으면 [-1]

해석

X가 아니면 섬이므로 일단 들어가서 연결되어있는 모든 섬을 방문하여 숫자를 더한 다음 결과 배열에 담는다.

방문했는지 여부를 알아야하니 visited 배열필요 

코드

import Foundation

extension String {
    subscript(_ index: Int) -> String {
        return String(self[self.index(self.startIndex, offsetBy: index)])
    } 
}


func solution(_ maps:[String]) -> [Int] {
    
    let n = maps.count
    let m = maps[0].count 
    var ans: [Int] = []
    
    var visited: [[Bool]] = [[Bool]](repeating: [Bool](repeating: false, count:m), count: n)

    let dx: [Int] = [0,0,-1,1]
    let dy: [Int] = [1,-1,0,0]
    
    func dfs(_ x:Int, _ y:Int) -> Int {
        
        visited[x][y] = true 
        var sum = 0 
        
        for i in 0..<4 {
            let nx = x + dx[i]
            let ny = y + dy[i]
            
            if nx < 0 || nx >= n || ny < 0 || ny >= m { continue }
            if visited[nx][ny] || maps[nx][ny] == "X" { continue }
            
            sum += dfs(nx,ny)
            
        }
        
        return Int(maps[x][y])! + sum
    }
    
    for i in 0..<n {
        for j in 0..<m {
            if maps[i][j] != "X" && !visited[i][j] {
                ans.append(dfs(i,j))
            }
        } 
            
    }
    
    return ans.isEmpty ? [-1] : ans.sorted()
}