[Swift/프로그래머스] 미로 탈출 ( 그리디 , BFS )

https://school.programmers.co.kr/learn/courses/30/lessons/159993

프로그래머스

 

문제 설명

1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.

미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.

---

제한사항

• 5 ≤ maps의 길이 ≤ 100
  • 5 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
    • S : 시작 지점
    • E : 출구
    • L : 레버
    • O : 통로
    • X : 벽
  • 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
  • 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.

 

일반적인 BFS 에서는 방문배열을 만들어서 확인하지만, 카운트도 해야하기에 따로 이동횟수를 저장한 배열을 사용해서 방문과 카운트를 동시에 체크해보았다.

 

import Foundation

func solution(_ maps:[String]) -> Int {
    // 일단 S에서 L로 가고
    // L에서 O로 이동하는 BFS 구현
    // 모든 곳을 -1 로 해두고 방문했을때 이전 곳 + 1을 해주자.
    var xSize = maps.count
    var ySize = maps[0].count
    
    var miro = [[Character]](repeating:[], count:xSize)
    var dp = [[Int]](repeating:[Int](repeating: 0, count: ySize ),count:xSize)
    
    var lab = (0,0)
    var start = (0,0)
    var end = (0,0)

    for (index,map) in maps.enumerated() {
        let arr = Array(map)
        miro[index] = arr
        for (indexy,char) in arr.enumerated() {
            if char == "X" {
                dp[index][indexy] = -1
            }else if char == "S" {
                start = (index,indexy)
            }else if char == "E" {
                end = (index,indexy)
            }else if char == "L" {
                lab = (index,indexy)
            }
        }
    }
    var dp2 = dp
    print(dp2)
    
    // S -> L
    let dx = [-1,0,1,0]
    let dy = [0,1,0,-1]
    var queue = [(Int,Int)]()
    queue.append(start)
    
    var findx = lab.0
    var findy = lab.1
    var flag = true
    while !queue.isEmpty && flag {
        let point = queue.removeFirst()
        let r = point.0
        let c = point.1
        for i in 0...3{
            let xr = r + dx[i]
            let yc = c + dy[i]
            if xr == findx && yc == findy {
                dp2[xr][yc]  = dp[r][c] + 1
                flag = false
            }
            if xr < 0 || xr >= xSize || yc < 0 || yc >= ySize{
                continue
            }
            if dp[xr][yc] == -1 || dp[xr][yc] != 0{
                continue
            }
            dp[xr][yc] = dp[r][c] + 1
            queue.append((xr,yc))
        }
    }
    
    if flag == true {
        return -1
    }
    
    
    
    queue = [(Int,Int)]()
    queue.append(lab)
    flag = true
    findx = end.0
    findy = end.1
    while !queue.isEmpty && flag {
        let point = queue.removeFirst()
        let r = point.0
        let c = point.1
        for i in 0...3{
            let xr = r + dx[i]
            let yc = c + dy[i]
            if xr == findx && yc == findy {
                dp2[xr][yc] = dp2[r][c] + 1
                return dp2[xr][yc]
            }
            if xr < 0 || xr >= xSize || yc < 0 || yc >= ySize{
                continue
            }
            if dp2[xr][yc] == -1 || dp2[xr][yc] != 0{
                continue
            }
            dp2[xr][yc] = dp2[r][c] + 1
            queue.append((xr,yc))
        }
    }
    
    return -1
}

 

너무 길다.

그래도 테스트케이스는 전부 통과했다.