[Programmers] Lv1. 공원산책(kotlin)

문제

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

 

 

예시

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

 

 

풀이

• 시뮬레이션 기법으로 하나씩 routes에 존재하는 방향, 거리만큼 이동한 후 확인하면 됩니다.
• isInside 함수를 미리 정의해두어 좌표 밖을 넘어가는 경우 체크
• 좌표 안에 있더라도 'X' 표기된 부분으로 장애물일 경우 체크
• 좌표 안에 있고 장애물도 아니라면 위치 이동

class Solution {
    val dir = mapOf("E" to (0 to 1), "N" to (-1 to 0), "W" to (0 to -1), "S" to (1 to 0))
    
    fun isInside(a: Int, b: Int, y:Int, x: Int) = 
        0<= a && a< y && 0<=b && b<x
    
    fun solution(park: Array<String>, routes: Array<String>): IntArray {
        var y = park.indexOfFirst{ it.contains("S")}
        var x = park[y].indexOf("S")
        
        routes.forEach{
            val input = it.split(" ")
            var dy = y
            var dx = x
            for(i in 0 until input.last().toInt()){
                dy += dir[input.first()]!!.first
                dx += dir[input.first()]!!.second
                
                if(isInside(dy, dx, park.size, park[0].length).not()) break
                if(park[dy][dx] == 'X') break
            }
            
            if(isInside(dy,dx, park.size, park[0].length) && park[dy][dx] != 'X'){
                y = dy
                x = dx
            }
        }
        
        
        return intArrayOf(y, x)
    }
}

 

 

참고

• 함수형 프로그래밍 언어인 kotlin의 특성에 맞게 함수를 많이 익히자 😽
• https://kotlinlang.org/api/latest/jvm/stdlib/kotlin.collections/index-of-first.html

indexOfFirst - Kotlin Programming Language